c++ opencv调用yolov5 onnx模型

最新推荐文章于 2024-05-25 21:42:30 发布

＋小子

最新推荐文章于 2024-05-25 21:42:30 发布

阅读量730

点赞数

分类专栏： pytorch 文章标签： opencv c++ 计算机视觉

本文链接：https://blog.csdn.net/qq15070/article/details/127419628

版权

// ReadOnnx.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//

#include <iostream>
#include<opencv2/opencv.hpp>
#include<opencv2/dnn/dnn.hpp>
#include<opencv2/highgui/highgui.hpp>
#include<vector>
#include <time.h>
using namespace std;
using namespace cv;
using namespace dnn;

inline float sigmoid(float x)
{
return 1.f / (1.f + exp(-x));
}

void drawPred(int classId, float conf, int left, int top, int right, int bottom, cv::Mat& frame,const std::vector<string>& classes)
{
   cv::rectangle(frame, cv::Point(left, top), cv::Point(right, bottom), cv::Scalar(0, 255, 0), 3);
   std::string label = cv::format("%.2f", conf);
   if (!classes.empty()) {
       CV_Assert(classId < (int)classes.size());
       label = classes[classId] + ": " + label;
   }
   int baseLine;
   cv::Size labelSize = cv::getTextSize(label, cv::FONT_HERSHEY_SIMPLEX, 0.5, 1, &baseLine);
   top = std::max(top, labelSize.height);
   cv::rectangle(frame, cv::Point(left, top - round(1.5 * labelSize.height)), cv::Point(left + round(1.5 * labelSize.width), top + baseLine), cv::Scalar(0, 255, 0), cv::FILLED);
   cv::putText(frame, label, cv::Point(left, top), cv::FONT_HERSHEY_SIMPLEX, 0.75, cv::Scalar(), 1);