鼓捣了一段时间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为自带的模型

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐