加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
exportOnnx.py 321 Bytes
一键复制 编辑 原始数据 按行查看 历史
# 这个是导出模型的工具,导出的模型在runs\detect\train\weights\best.onnx位置
from ultralytics import YOLO
# 加载模型
#model = YOLO('yolov8n.pt') # 加载官方模型
model = YOLO('runs/detect/train/weights/best.pt') # 使用自己训练出来的模型
# 导出模型
model.export(format='onnx')
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化