yolov8之导出onnx(二)

慈云数据 2024-03-13 技术支持 44 0

前面(一)我们给出了导出的batCh是固定的,接下来我们实现动态导出:

 

导出的主代码demo.py:

IMPort os
from ultralytics import YOLO
model = YOLO("yolov8s.yaml")
model=YOLO("../pretrained_model/yolov8s.pt")
#success=model.export(format="onnx")
success=model.export(format="onnx", half=False, dynamic=True, opset=17)
print("demo")

注意:half与dynamic必须二选一才行!

 

 

导出的onnx:

微信扫一扫加客服

微信扫一扫加客服

点击启动AI问答
Draggable Icon