Update OpenVINO CI for Python 3.12 (#16420)

pull/16421/head
Glenn Jocher 4 months ago committed by GitHub
parent 5105dcc297
commit 486d4cbe54
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 1
      tests/test_exports.py
  2. 4
      ultralytics/utils/benchmarks.py

@ -32,7 +32,6 @@ def test_export_onnx():
YOLO(file)(SOURCE, imgsz=32) # exported model inference YOLO(file)(SOURCE, imgsz=32) # exported model inference
@pytest.mark.skipif(checks.IS_PYTHON_3_12, reason="OpenVINO not supported in Python 3.12")
@pytest.mark.skipif(not TORCH_1_13, reason="OpenVINO requires torch>=1.13") @pytest.mark.skipif(not TORCH_1_13, reason="OpenVINO requires torch>=1.13")
def test_export_openvino(): def test_export_openvino():
"""Test YOLO exports to OpenVINO format for model inference compatibility.""" """Test YOLO exports to OpenVINO format for model inference compatibility."""

@ -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

Loading…
Cancel
Save