YOLOv8模型转换pt-＞onnx(附上代码)

作者在用fastdeploy部署YOLOv8模型时要将pt文件转换成onnx

于是可以在官方文件中找到对应方法链接：Export - Ultralytics YOLOv8 Docs

笔者在创建的虚拟环境中安装ultralytics

命令行为

pip install ultralytics

 然后笔者新创建了python文件，其中代码示例如下

from ultralytics import YOLO

# 加载模型
model = YOLO('yolov8n.pt')  # 加载官方模型（示例）
model = YOLO('D:/Web page download/best2.pt')  # 加载自定义训练模型（示例）

# 导出模型
model.export(format='onnx')

 注意自己训练好的pt文件的地址和加载的官方模型

最后就可以在对应文件夹下面找到转换成功的onnx文件