yolov5 6.1同时支持Opencv dnn、TensorRT、Edge TPU和OpenVINO模块部署,在工程实现上就方便多了。
首先要将pt文件导出为onnx模型文件
运行export.py,--weights参数后加pt模型文件路径,结束后得到.onnx后缀的文件。
python export.py --weights yolov5s.pt --include torchscript onnx
接下来就是opencv dnn的实现了,官方给了参考Github,不过如果是自己训练的模型,需要注意的是:
1、
将74行 const int dimensions = 85;
改为 const int dimensions = 5+类别数;
2、
将110行 data += 85;
改为 data += dimensions;
---------------------------------------------------------------------------
当模型的imgsz改变时,下面的值也要做相应改变
const float INPUT_WIDTH = 640.0;
const float INPUT_HEIGHT = 640.0;
const int rows = 25200;用Netron Viewer也可以看出来模型输出大小
下面自己C++封装了一个类:
头文件:
#include <fstream>
#include <sstream>
#include <iostream>
#include <opencv2/dnn.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/highgui.hp

这篇博客介绍了如何将Yolov5模型转换为ONNX格式,并在C++中使用OpenCV进行部署。作者详细展示了如何修改源代码以适配自定义的类别数,并提供了C++封装的YOLO类。然而,作者发现使用OpenCV的DNN模块在CPU上的运行速度并不如原算法快。此外,文中还提及了Yolov5在OpenVINO平台上的实现选项。


3641

被折叠的 条评论
为什么被折叠?



