yolov5 pt转onnx

参考：.

Windows搭建yolo5训练环境，及pt文件转为onnx导出

检查显卡配置，查询可支持的cuda版本

win左下角搜索cmd打开命令行窗口，输入nvidia-smi

 

一：将PT模型文件转onnx：

1、用到export.py

安装依赖包

pip3 install onnx

pip3 install coremltools==4.0

2、export.py中的配置

文件路径

进行修改：

 

 运行export.py,

生成best.onnx