xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use new OpenVINO API instead of deprecated (#736)

Browse Source
This commit is contained in:
Adrian Boguszewski 2023-02-02 11:59:58 +01:00 committed by GitHub
parent 522f1937ed
commit b83374b42d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements.txt
View File

@ -31,7 +31,7 @@ seaborn>=0.11.0
# scikit-learn==0.19.2 # CoreML quantization # scikit-learn==0.19.2 # CoreML quantization
# tensorflow>=2.4.1 # TF exports (-cpu, -aarch64, -macos) # tensorflow>=2.4.1 # TF exports (-cpu, -aarch64, -macos)
# tensorflowjs>=3.9.0 # TF.js export # tensorflowjs>=3.9.0 # TF.js export
# openvino-dev # OpenVINO export # openvino-dev>=2022.1 # OpenVINO export
# Extras -------------------------------------- # Extras --------------------------------------
ipython # interactive notebook ipython # interactive notebook

4
ultralytics/yolo/engine/exporter.py
View File

@ -320,9 +320,9 @@ class Exporter:
def _export_openvino(self, prefix=colorstr('OpenVINO:')): def _export_openvino(self, prefix=colorstr('OpenVINO:')):
# YOLOv8 OpenVINO export # YOLOv8 OpenVINO export
check_requirements('openvino-dev') # requires openvino-dev: https://pypi.org/project/openvino-dev/ check_requirements('openvino-dev') # requires openvino-dev: https://pypi.org/project/openvino-dev/
import openvino.inference_engine as ie # noqa import openvino.runtime as ov # noqa
LOGGER.info(f'\n{prefix} starting export with openvino {ie.__version__}...') LOGGER.info(f'\n{prefix} starting export with openvino {ov.__version__}...')
f = str(self.file).replace(self.file.suffix, f'_openvino_model{os.sep}') f = str(self.file).replace(self.file.suffix, f'_openvino_model{os.sep}')
f_onnx = self.file.with_suffix('.onnx') f_onnx = self.file.with_suffix('.onnx')