本文主要是介绍Darknet Yolo + VS +QtGui 图形界面可视化输出 分类输出检测目标数目 简单的计数器实现 yolo动态链接库,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
本文的图形界面可视化输出是基于下述文章来写的:
参考文章:QtGUI+YOLOv4+VS2017 《图形界面+目标检测 cpu版本&gpu版本》_scut_lrr的博客-CSDN博客_qt yolo
QtGUI+YOLOv4+VS2017 《图形界面+目标检测 cpu版本&gpu版本》
实现思路,利用vector容器,找到绘制矩形框的代码处,根据绘制情况,来进行针对不同ojb_name的计数操作。
具体实现:
1.将上述项目部署完成之后,打开common.h文件,找到 draw_boxes函数。
在这里为了不影响代码原有功能,根据draw_boxes函数,新建draw_boxes2函数,并增加两个新的形参,vector容器,和int类型的计数器.
(新建的形参要放在非末尾两个位置的任意位置)
2.然后在for循环里
for (auto &i : result_vec) {count+=1;//保证每次识别到对象绘制矩形框之前,计数器加一cv::Scalar color(0, 255, 0);cv::rectangle(mat_img, cv::Rect(i.x, i.y, i.w, i.h), color, 1);std::string obj_name = obj_names[i.obj_id]; std::string obj_id = std::to_string(i.obj_id); zz[i.obj_id] += 1;//将obj_id作为每个类的索引,每识别到此次绘制的类别,+1;cv::Size const text_size = getTextSize(obj_name, cv::FONT_HERSHEY_SIMPLEX, 0.5, 1, 0);int const max_width = (text_size.width > i.w + 2) ? text_size.width : (i.w + 2);cv::rectangle(mat_img, cv::Point2f(std::max((int)i.x, 0), std::max((int)i.y, 0)),cv::Point2f(std::min((int)i.x, mat_img.cols), std::min((int)i.y, mat_img.rows)),color, 0.8, 0.8, 0);putText(mat_img, obj_id, cv::Point2f(i.x, i.y), cv::FONT_HERSHEY_COMPLEX_SMALL, 1, cv::Scalar(0, 0, 0), 1);}std::string spend = QS(QString::number(spendTime, 10, 4));
3.打开QGuiDemo.cpp 找到输出函数openPictureSlot
vector<int> a = {0,0};//定义一个vector容器和int类型数
int c = 0;draw_boxes2(a,c,inputImage, result_vec, obj_names, spent.count()); //调用draw_boxes2this->textLabel->setText("a:" + QString::number(a[0]) + " " + "b:" + QString::number(a[1]) + " "+"sum:"+ QString::number(c) );
//目前还没想到更好的 判断数组是否越界的办法,只能通过手动控制来输出
//比如这里有几个类 就显示多少,比如我这里有两个类,那么就是a[0]和a[1],如果是3个类的话就加一个a[2],
//注意这里,在label上显示数字需要使用 QString::number();
4.显示结果
这里是拿 QtGuiDemo项目来做的,其他使用YOLO动态链接库的项目应该都可以按照这个办法来实现。
再附一张图,当时忘记输出sum了,加上就好:
这篇关于Darknet Yolo + VS +QtGui 图形界面可视化输出 分类输出检测目标数目 简单的计数器实现 yolo动态链接库的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!