16 lines
369 B
Python
16 lines
369 B
Python
from ultralytics import YOLO
|
|
import openvino as ov
|
|
'''
|
|
加载文件失败
|
|
'''
|
|
# 加载 OpenVINO 模型
|
|
core = ov.Core()
|
|
compiled_model = core.compile_model("best_s.xml", "CPU")
|
|
|
|
# 通过 YOLO 包装推理
|
|
model = YOLO("best_s.xml") # 使用 YOLO 接口
|
|
model.model = compiled_model # 替换为 OpenVINO 模型
|
|
|
|
|
|
# 执行推理
|
|
results = model.predict("image.jpg") |