[C++]使用纯opencv部署yolov12目标检测onnx模型

devtools/2025/2/22 20:06:20/

yolov12官方框架:sunsmarterjie/yolov12

【算法介绍】

在C++中使用纯OpenCV部署YOLOv12进行目标检测是一项具有挑战性的任务,因为YOLOv12通常是用PyTorch等深度学习框架实现的,而OpenCV本身并不直接支持加载和运行PyTorch模型。然而,你可以通过一些间接的方法来实现这一目标,比如将PyTorch模型转换为ONNX格式,然后使用OpenCV的DNN模块加载ONNX模型。

以下是一个大致的步骤指南,用于在C++中使用OpenCV部署YOLOv12(假设你已经有了YOLOv12的ONNX模型):

  1. 安装依赖
    • 确保你的开发环境已经安装了OpenCV 4.8(带有DNN模块)和必要的C++编译器。
  2. 准备模型
    • YOLOv12模型从PyTorch转换为ONNX格式。这通常涉及使用PyTorch的torch.onnx.export函数。
    • 确保你有YOLOv12的ONNX模型文件、配置文件(描述模型架构)和类别名称文件。
  3. 编写C++代码
    • 使用OpenCV的DNN模块加载ONNX模型。
    • 预处理输入图像(如调整大小、归一化等),以符合模型的输入要求。
    • 将预处理后的图像输入到模型中,并获取检测结果。
    • 对检测结果进行后处理,包括解析输出、应用非极大值抑制(NMS)和绘制边界框。
  4. 编译和运行
    • 使用C++编译器(如g++)编译你的代码。
    • 运行编译后的程序,输入图像或视频,并观察目标检测结果。

需要注意的是,由于YOLOv12是一个复杂的模型,其输出可能包含多个层的信息(如特征图、置信度、边界框坐标等),因此你需要仔细解析模型输出,并根据YOLOv12的具体实现进行后处理。

此外,由于OpenCV的DNN模块对ONNX的支持可能有限,某些YOLOv12的特性(如自定义层、特定的激活函数等)可能无法在OpenCV中直接实现。在这种情况下,你可能需要寻找替代方案,如使用其他深度学习库(如TensorRT、ONNX Runtime等)来加载和运行模型,并通过C++接口与这些库进行交互。

总之,在C++中使用纯OpenCV部署YOLOv12是一项具有挑战性的任务,需要深入理解YOLOv12的模型架构、OpenCV的DNN模块以及ONNX格式。如果你不熟悉这些领域,可能需要花费更多的时间和精力来学习和解决问题。

【效果展示】

 

【测试环境】

vs2019
cmake==3.30.0
opencv==4.8.0

【调用代码】

#include <iostream>
#include<opencv2/opencv.hpp>#include<math.h>
#include "yolov12.h"
#include<time.h>
#define  VIDEO_OPENCV //if define, use opencv for video.using namespace std;
using namespace cv;
using namespace dnn;template<typename _Tp>
int yolov12(_Tp& cls,Mat& img,string& model_path)
{Net net;if (cls.ReadModel(net, model_path, false)) {cout << "read net ok!" << endl;}else {return -1;}//生成随机颜色vector<Scalar> color;srand(time(0));for (int i = 0; i < 80; i++) {int b = rand() % 256;int g = rand() % 256;int r = rand() % 256;color.push_back(Scalar(b, g, r));}vector<OutputSeg> result;if (cls.Detect(img, net, result)) {DrawPred(img, result, cls._className, color);}else {cout << "Detect Failed!" << endl;}system("pause");return 0;
}template<typename _Tp>
int video_demo(_Tp& cls, string& model_path)
{vector<Scalar> color;srand(time(0));for (int i = 0; i < 80; i++) {int b = rand() % 256;int g = rand() % 256;int r = rand() % 256;color.push_back(Scalar(b, g, r));}vector<DetecionResult> result;cv::VideoCapture cap("car.mp4");if (!cap.isOpened()){std::cout << "open capture failured!" << std::endl;return -1;}Mat frame;
#ifdef VIDEO_OPENCVNet net;if (cls.ReadModel(net, model_path, true)) {cout << "read net ok!" << endl;}else {cout << "read net failured!" << endl;return -1;}#elseif (cls.ReadModel(model_path, true)) {cout << "read net ok!" << endl;}else {cout << "read net failured!" << endl;return -1;}#endifwhile (true){cap.read(frame);if (frame.empty()){std::cout << "read to end" << std::endl;break;}result.clear();
#ifdef VIDEO_OPENCVif (cls.Detect(frame, net, result)) {DrawPred(frame, result, cls._className, color, true);}
#elseif (cls.OnnxDetect(frame, result)) {DrawPred(frame, result, cls._className, color, true);}
#endifint k = waitKey(10);if (k == 27) { //esc break;}}cap.release();system("pause");return 0;
}int main() {string detect_model_path = "./yolov12n.onnx";Yolov12 detector;video_demo(detector, detect_model_path);
}

视频演示:

使用纯opencv部署yolov12目标检测onnx模型演示源码+模型_哔哩哔哩_bilibili


http://www.ppmy.cn/devtools/161015.html

相关文章

cuda安装

cuda WSL2急速搭建CUDA体验环境_wsl2 cuda-CSDN博客 cudnn cuDNN Archive | NVIDIA Developer pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

PyTorch 源码学习:阅读经验 代码结构

分享自己在学习 PyTorch 源码时阅读过的资料。本文重点关注阅读 PyTorch 源码的经验和 PyTorch 的代码结构。因为 PyTorch 不同版本的源码实现有所不同&#xff0c;所以笔者在整理资料时尽可能按版本号升序&#xff0c;版本号见标题前[]。最新版本的源码实现还请查看 PyTorch 仓…

Docker安装Quickwit搜索引擎

介绍 Quickwit 是首个以亚秒级延迟直接在云存储上执行复杂搜索和分析查询的引擎。它由 Rust 及其分离的计算和存储架构提供支持&#xff0c;旨在节省资源、易于操作并可扩展到 PB 级数据。 Quickwit 非常适合日志管理、分布式跟踪以及通常不可变的数据&#xff08;例如对话数…

本地部署DeepSeek的硬件配置建议

本地部署DeepSeek的硬件配置需求因模型参数规模和部署工具不同而有所差异&#xff0c;以下是综合多个来源的详细要求&#xff1a; 1. 基础配置&#xff08;适用于7B参数模型&#xff09; 内存&#xff1a;最低8GB&#xff0c;推荐16GB及以上&#xff1b;若使用Ollama工具&…

Datawhale Ollama教程笔记5

Dify 接入 Ollama 部署的本地模型 Dify 支持接入 Ollama 部署的大型语言模型推理和 embedding 能力。 快速接入 下载 Ollama 访问 Ollama 安装与配置&#xff0c;查看 Ollama 本地部署教程。 运行 Ollama 并与 Llama 聊天 ollama run llama3.1Copy to clipboardErrorCopied …

安装海康威视相机SDK后,catkin_make其他项目时,出现“libusb_set_option”错误的解决方法

硬件&#xff1a;雷神MIX G139H047LD 工控机 系统&#xff1a;ubuntu20.04 之前运行某项目时&#xff0c;处于正常状态。后来由于要使用海康威视工业相机&#xff08;型号&#xff1a;MV-CA013-21UC&#xff09;&#xff0c;便下载了并安装了该相机的SDK&#xff0c;之后运行…

go WEB框架

推荐选型 https://chat.deepseek.com/a/chat/s/e6061607-8f33-4768-a5f0-8970cb1ffefd echo github&#xff1a;https://github.com/labstack/echo wiki&#xff1a;https://echo.labstack.com/docs/quick-start block&#xff1a;https://blog.csdn.net/qq_38105536/artic…

HTML Canvas clip 深入全面讲解

在HTML5的<canvas>元素中&#xff0c;裁剪路径&#xff08;clipping path&#xff09;是一个非常强大的功能&#xff0c;它允许你定义一个区域&#xff0c;在该区域内进行绘制&#xff0c;而区域外的内容将不会被显示。通过合理使用裁剪路径&#xff0c;可以实现复杂的图…