目标检测中遇到需要统计目标个数的情况,所以有了一下操作:

首先来看一下效果

 

由此图可以看到:目标检测框在label前添加了序号(因为我最终的目标只有玉米,所以没有考虑目标分类排序),并且在图像的左上方还添加了文本,统计各个类别的物体的个数。

接下来,我们一步一步来通过改写代码实现以上两个功能:

红线所指的方向就是我们需要改的两个文件。

首先是plotting.py中的box_label函数中的四个位置,按我所标注的添加即可 。

这里是detect.py中需要修改的五个地方,最下面的一条语句我给复制下来,可以直接粘贴 

        cv2.putText(im0, log_string, (5, 50), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (0, 0, 255), 2)

 这样就可以实现最上面的功能了,希望对大家能有帮助。。。

发现大家都已经用新版本的yolov8了,这个不怎么适用了。

如果很急可以私信我。。。

sdau

点击阅读全文
Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐