Bringing OpenVINO updates from YOLOv8 to YOLOv5 (#11912)
Bringing OpenVINO updates from yolov8 to volov5 Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
1 parent
d43d87666c
commit
34c21874f8
5 files changed
+13
-13
No files matched your search
@@ -207,7 +207,7 @@ def export_onnx(model, im, file, opset, dynamic, simplify, prefix=colorstr('ONNX
|
||||
@try_export
|
||||
def export_openvino(file, metadata, half, int8, data, prefix=colorstr('OpenVINO:')):
|
||||
# YOLOv5 OpenVINO export
|
||||
check_requirements('openvino-dev>=2022.3') # requires openvino-dev: https://pypi.org/project/openvino-dev/
|
||||
check_requirements('openvino-dev>=2023.0') # requires openvino-dev: https://pypi.org/project/openvino-dev/
|
||||
import openvino.runtime as ov # noqa
|
||||
from openvino.tools import mo # noqa
|
||||
|
||||
@@ -216,7 +216,7 @@ def export_openvino(file, metadata, half, int8, data, prefix=colorstr('OpenVINO:
|
||||
f_onnx = file.with_suffix('.onnx')
|
||||
f_ov = str(Path(f) / file.with_suffix('.xml').name)
|
||||
if int8:
|
||||
check_requirements('nncf')
|
||||
check_requirements('nncf>=2.4.0') # requires at least version 2.4.0 to use the post-training quantization
|
||||
import nncf
|
||||
import numpy as np
|
||||
from openvino.runtime import Core
|
||||
|
||||
Reference in new issue
Block a user