yolov8之导出onnx(二)

前面(一)我们给出了导出的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:

猜你喜欢

转载自blog.csdn.net/yangjinyi1314/article/details/128788184
今日推荐