前面(一)我们给出了导出的batch是固定的,接下来我们实现动态导出:
文章来源:https://www.toymoban.com/news/detail-567575.html
导出的主代码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:文章来源地址https://www.toymoban.com/news/detail-567575.html
到了这里,关于yolov8之导出onnx(二)的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!