目标检测中遇到需要统计目标个数的情况,所以有了一下操作:
首先来看一下效果
由此图可以看到:目标检测框在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