如何在windows系统下将yolov5的pt模型导出为onnx模型

前言

最近在做本科毕业设计，要求是在树莓派上部署yolo算法来实现火灾检测，在网上查了很多资料，最后选择用yolov5s模型先试着在树莓派上部署，看下效果如何，由于从大佬那里拿到了yolov5火灾检测模型，但想要将它移植到树莓派上第一步要把pt模型转换成onnx模型，原因我想大概是因为树莓派算力不如pc，因此要转换成合适的模型进行部署。一开始我觉得转换模型应该是很简单的一个事情，但是在进行的过程中踩了很多坑，走了很多弯路，因此在这里总结一下，也希望有相同需求的朋友能从这里获得帮助，ps:因为本人水平有限所以能帮到大家很高兴。

第一个问题

在进行模型转化时从网上找的代码是这样：

python export.py --weights ‘weights/last.pt’ --batch-size 1 --img_size 320

大概内容是调用export.py这个文件来进行模型的转化，但是将代码复制到项目命令行里运行后产生如下报错：

我看了下报错的原因大概是语法错误，后来在网上查找后发现其实这个命令可以不带后面的–batch-size和–img-size，如果要带的话这里的–img_size应该改为–img-size。

第二个问题

将后面的参数删掉以后发现还是不能正常运行，后来经过分析后发现是因为这里的’weights/last.pt’应该是你项目里这个.pt模