作者在用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文件

Logo

欢迎加入 MCP 技术社区!与志同道合者携手前行,一同解锁 MCP 技术的无限可能!

更多推荐