opencv调用yolov7 yolov7 c++ yolov7转onnx opencv调用yolov7 onnx

一、YOLOV7主要贡献:

        主要是现有的一些trick的集合以及模块重参化和动态标签分配策略,最终在 5 FPS 到 160 FPS 范围内的速度和准确度都超过了所有已知的目标检测器。

        当前目标检测主要的优化方向:更快更强的网络架构;更有效的特征集成方法;更准确的检测方法;更精确的损失函数;更有效的标签分配方法;更有效的训练方法。

二、参考代码:

1、C++参考地址:
https://github.com/UNeedCryDear/yolov7-opencv-dnn-cpp

2、yolov7的版本:

https://github.com/WongKinYiu/yolov7

三、应用说明:

        在项目部署过程中,为了脱离pytorch而只使用C++调用,我参考了市面上N多解决办法,最终把程序调试通过,调用过程是先把yolov7.pt转化为yolov7.onnx,之后再通过opencv dnn来调用onnx。

四、调用流程说明:

1、环境条件:

windows10、vs2015、opencv4.5.5、python3.8

2、下载 python版本的yolov7,导出onnx

下载地址:https://github.com/WongKinYiu/yolov7

目录为yolov7-main:

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

注意:之前我安装的pytorch gpu版本一直没有通过,后来把 requirements.txt中的所有库都卸载了,重新执行上面的指令,导出onnx时才成功

导出onnx

进入到yolov7-main目录下,输入:

python export.py --weights ./yolov7.pt --grid --end2end --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640

中间有些小错误,不用理睬:

3、下载opencv c++调用程序:

 下载:https://github.com/UNeedCryDear/yolov7-opencv-dnn-cpp

打开vs2015建立新测试工程:

4、添加模型路径后,直接运行即可,运行结果如下:

 

 

四、源代码

为了大家能够都方便的使用yolov7 c++程序,现将程序代码提交:

百度网盘地址:

链接:https://pan.baidu.com/s/1-01EuH4oH_3HPo4H50g3wQ 
提取码:bruh 
 

QQ交流:187100248.

 

  • 12
    点赞
  • 86
    收藏
    觉得还不错? 一键收藏
  • 29
    评论
要在C++调用OpenCVYolov7,需要以下步骤: 1. 下载和安装OpenCVYolov7。可以从官方网站下载OpenCV,也可以从GitHub上下载Yolov7。 2. 加载Yolov7模型。在C++中,可以使用dnn::readNetFromDarknet函数来加载Yolov7模型。 3. 加载图像并进行预处理。在C++中,可以使用cv::imread函数来加载图像,并使用cv::dnn::blobFromImage函数来进行预处理。 4. 执行前向传递。在C++中,可以使用cv::dnn::Net::forward函数来执行前向传递。 5. 解析输出并绘制边界框。在C++中,可以使用cv::dnn::NMSBoxes函数来解析输出,并使用cv::rectangle函数来绘制边界框。 以下是一个简单的示例代码,展示了如何在C++调用OpenCVYolov7: ``` #include <opencv2/opencv.hpp> #include <opencv2/dnn.hpp> int main() { // 加载Yolov7模型 cv::dnn::Net net = cv::dnn::readNetFromDarknet("yolov7.cfg", "yolov7.weights"); // 加载图像并进行预处理 cv::Mat image = cv::imread("image.jpg"); cv::Mat blob = cv::dnn::blobFromImage(image, 1 / 255.0, cv::Size(416, 416), cv::Scalar(), true, false); // 执行前向传递 net.setInput(blob); cv::Mat output = net.forward(); // 解析输出并绘制边界框 std::vector<cv::Mat> detections(output.size[0]); for (int i = 0; i < output.size[0]; ++i) { cv::Mat detection(output.size[1], output.size[2], CV_32F, output.ptr<float>(i)); detections[i] = detection; } std::vector<int> indices; cv::dnn::NMSBoxes(bboxes, confidences, 0.5, 0.4, indices); for (size_t i = 0; i < indices.size(); ++i) { int idx = indices[i]; cv::Rect bbox = bboxes[idx]; float confidence = confidences[idx]; cv::rectangle(image, bbox, cv::Scalar(0, 255, 0), 2); } cv::imshow("Yolov7", image); cv::waitKey(0); return 0; } ```

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 29
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值