|
|
@ -97,8 +97,8 @@ def benchmark( |
|
|
|
assert MACOS or LINUX, "CoreML and TF.js export only supported on macOS and Linux" |
|
|
|
assert MACOS or LINUX, "CoreML and TF.js export only supported on macOS and Linux" |
|
|
|
assert not IS_RASPBERRYPI, "CoreML and TF.js export not supported on Raspberry Pi" |
|
|
|
assert not IS_RASPBERRYPI, "CoreML and TF.js export not supported on Raspberry Pi" |
|
|
|
assert not IS_JETSON, "CoreML and TF.js export not supported on NVIDIA Jetson" |
|
|
|
assert not IS_JETSON, "CoreML and TF.js export not supported on NVIDIA Jetson" |
|
|
|
if i in {3, 5}: # CoreML and OpenVINO |
|
|
|
if i in {5}: # CoreML |
|
|
|
assert not IS_PYTHON_3_12, "CoreML and OpenVINO not supported on Python 3.12" |
|
|
|
assert not IS_PYTHON_3_12, "CoreML not supported on Python 3.12" |
|
|
|
if i in {6, 7, 8}: # TF SavedModel, TF GraphDef, and TFLite |
|
|
|
if i in {6, 7, 8}: # TF SavedModel, TF GraphDef, and TFLite |
|
|
|
assert not isinstance(model, YOLOWorld), "YOLOWorldv2 TensorFlow exports not supported by onnx2tf yet" |
|
|
|
assert not isinstance(model, YOLOWorld), "YOLOWorldv2 TensorFlow exports not supported by onnx2tf yet" |
|
|
|
if i in {9, 10}: # TF EdgeTPU and TF.js |
|
|
|
if i in {9, 10}: # TF EdgeTPU and TF.js |
|
|
|