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

cpu推理onnx指令：

D:\anaconda3\envs\pytorch\python.exe D:/deep_learning/yolov5-master/detect.py --weights side_defect.onnx --source D:\dee
p_learning\side_defect_val_img --device cpu