opencv dnn模块实现Yolov5_6.1

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

yolov5 6.1同时支持Opencv dnn、TensorRT、Edge TPU和OpenVINO模块部署,在工程实现上就方便多了。

Yolov5_6.1

首先要将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
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值