目录
YOLOv5s-v3.1模型转换
因为U版YOLO的作者用pytorch实现,yolov5的模型是torch格式的pt。要使用ncnn部署YOLOv5需要将模型进行转换,大概步骤是pt2onnx,onnx2ncnn。
因为只有yolovs-v3.1版本正常测试通过了,固仅针对此进行记录。高版本的yolov5初试未成功,未细加研究,之后有时间再进行深入研究。。。。
模型的转换主要参考:详细记录u版YOLOv5目标检测ncnn实现 - 知乎
第一步:pt2onnx
准备一个已经训练好了的yolov5s-v3.1.pt模型,pt2onnx我就用的YOLOv5-v3.1版本源码中的export.py进行的转换
生成yolov5s-v3.1.onnx,并用onnxsim简化一下生成yolov5s-v3.1-sim.onnx
$ python models/export.py --weights yolov5s-v3.1.pt --img 640 --batch 1
$ python -m onnxsim yolov5s-v3.1.onnx yolov5s-v3.1-sim.onnx
这一步还比较顺利~