mirror of
https://github.com/THU-MIG/yolov10.git
synced 2025-05-23 21:44:22 +08:00
OpenVino 2024.0.0 (#8726)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
parent
f5d883ba37
commit
b53a3989ab
@ -98,7 +98,7 @@ dev = [
|
|||||||
export = [
|
export = [
|
||||||
"onnx>=1.12.0", # ONNX export
|
"onnx>=1.12.0", # ONNX export
|
||||||
"coremltools>=7.0; platform_system != 'Windows' and python_version <= '3.11'", # CoreML supported on macOS and Linux
|
"coremltools>=7.0; platform_system != 'Windows' and python_version <= '3.11'", # CoreML supported on macOS and Linux
|
||||||
"openvino>=2023.3; python_version <= '3.11'", # OpenVINO export
|
"openvino>=2024.0.0", # OpenVINO export
|
||||||
"tensorflow<=2.13.1; python_version <= '3.11'", # TF bug https://github.com/ultralytics/ultralytics/issues/5161
|
"tensorflow<=2.13.1; python_version <= '3.11'", # TF bug https://github.com/ultralytics/ultralytics/issues/5161
|
||||||
"tensorflowjs>=3.9.0; python_version <= '3.11'", # TF.js export, automatically installs tensorflow
|
"tensorflowjs>=3.9.0; python_version <= '3.11'", # TF.js export, automatically installs tensorflow
|
||||||
]
|
]
|
||||||
|
@ -411,7 +411,7 @@ class Exporter:
|
|||||||
@try_export
|
@try_export
|
||||||
def export_openvino(self, prefix=colorstr("OpenVINO:")):
|
def export_openvino(self, prefix=colorstr("OpenVINO:")):
|
||||||
"""YOLOv8 OpenVINO export."""
|
"""YOLOv8 OpenVINO export."""
|
||||||
check_requirements("openvino>=2023.3") # requires openvino: https://pypi.org/project/openvino/
|
check_requirements("openvino>=2024.0.0") # requires openvino: https://pypi.org/project/openvino/
|
||||||
import openvino as ov
|
import openvino as ov
|
||||||
|
|
||||||
LOGGER.info(f"\n{prefix} starting export with openvino {ov.__version__}...")
|
LOGGER.info(f"\n{prefix} starting export with openvino {ov.__version__}...")
|
||||||
|
@ -190,7 +190,7 @@ class AutoBackend(nn.Module):
|
|||||||
# OpenVINO
|
# OpenVINO
|
||||||
elif xml:
|
elif xml:
|
||||||
LOGGER.info(f"Loading {w} for OpenVINO inference...")
|
LOGGER.info(f"Loading {w} for OpenVINO inference...")
|
||||||
check_requirements("openvino>=2023.3")
|
check_requirements("openvino>=2024.0.0")
|
||||||
import openvino as ov
|
import openvino as ov
|
||||||
|
|
||||||
core = ov.Core()
|
core = ov.Core()
|
||||||
|
Loading…
x
Reference in New Issue
Block a user