21xrx.com
2024-11-22 02:44:14 Friday
登录
文章检索 我的文章 写文章
C++使用OpenCV部署YOLOv5模型
2023-07-02 22:47:32 深夜i     --     --
C++ OpenCV YOLOv5 部署 模型

近年来,物体检测技术在计算机视觉领域得到了广泛的关注和研究。其中,YOLO系列模型就是一个非常出色的物体检测模型。在这篇文章中,我们将讨论如何在C++中使用OpenCV部署YOLOv5模型。

首先,我们需要下载YOLOv5模型的权重文件。这些权重文件可以从GitHub上的开源项目中下载。下载完毕后,我们需要下载OpenCV库。OpenCV是一个广泛使用的计算机视觉库,提供了许多有用的函数和算法。它可以帮助我们在C++中轻松加载和处理图像。

接下来,我们需要编写一些C++代码来加载模型并运行推理。以下是一个使用OpenCV和YOLOv5模型进行物体检测的示例代码:


#include <iostream>

#include "opencv2/opencv.hpp"

using namespace cv;

using namespace std;

int main(int argc, char **argv) {

  // Load the YOLOv5 model

  string modelConfiguration = "yolov5.cfg";

  string modelWeights = "yolov5.weights";

  dnn::Net net = dnn::readNetFromDarknet(modelConfiguration, modelWeights);

  // Load the input image

  Mat image = imread("test.jpg");

  // Create a 4D blob from the input image and set the input of the network to the blob

  Mat blob = dnn::blobFromImage(image, 1 / 255.0, Size(416, 416), Scalar(0, 0, 0), true, false);

  net.setInput(blob);

  // Run forward pass to get output of the output layers

  vector<Mat> outputs;

  net.forward(outputs, net.getUnconnectedOutLayersNames());

  // Process each output layer

  for (int i = 0; i < outputs.size(); i++) {

    Mat detectionMat = outputs[i];

    // Loop over all detections and draw bounding boxes around the objects

    for (int j = 0; j < detectionMat.rows; j++) {

      float score = detectionMat.at<float>(j, 5);

      if (score > 0.5) {

        float x = detectionMat.at<float>(j, 0) * image.cols;

        float y = detectionMat.at<float>(j, 1) * image.rows;

        float width = detectionMat.at<float>(j, 2) * image.cols;

        float height = detectionMat.at<float>(j, 3) * image.rows;

        float x1 = x - width / 2;

        float y1 = y - height / 2;

        rectangle(image, Point(x1, y1), Point(x1 + width, y1 + height), Scalar(0, 255, 0), 2);

      }

    }

  }

  // Show the image with the detected objects

  imshow("YOLOv5 object detection", image);

  waitKey(0);

}

上面的代码首先加载了YOLOv5模型的权重文件,然后加载测试图片。然后,它将图像转换为4D的blob,然后将其设置为网络的输入。接下来,网络运行推理并得到输出层的输出。最后,代码循环遍历每个检测,并在图像中绘制由该检测生成的边界框。

最后,当我们编译并运行这个程序时,我们将得到一个显示检测到的对象的图像。您可以调整代码以便更改模型和输入图像的参数,以实现不同的物体检测任务。

总之,使用OpenCV和YOLOv5模型在C++中进行物体检测是非常简单和有用的。我们只需要加载模型、加载图像、处理输出并绘制边界框。这使得利用YOLOv5模型进行计算机视觉任务变得更加简单和快速。

  
  

评论区

{{item['qq_nickname']}}
()
回复
回复