鼓捣了一段时间Yolv5了,现在将如何通过onnx将Yolv5模型完成c++部署,记录下来,便于交流讨论。
一、实现环境
1、pytorch 1.10
2、python 3.9
3、opencv4.5.4
4、yolv5 v6.0
二、将自己的训练模型或自带的yolov5s.pt模型通过yolv5自带的 export.py,导出yolov5s.onnx模型
三、下载部署文件 https://github.com/Hexmagic/ONNX-yolov5
四、根据下图设置并编译,注意需要配置opencv的路径
五、调试和验证,best.onnx为自己训练的模型,yolv5s.onnx为自带的模型