目标检测YOLOV7 添加计数功能

开学我就研二下了,感叹时光飞逝。因为好多人在评论里问,所以出一篇!注:本文章应该是适用所有YOLO系列。

1. 只有一种类别

修改detect.py文件,找到if save_img,下面添加就可以了

目标检测YOLOV7 添加计数功能_第1张图片

cv2.putText(im0, f"{n} {names[int(c)]}{'s' * (n > 1)}", (5, 50), cv2.FONT_HERSHEY_SIMPLEX, 1.2,(0, 0, 255), 2)   # 只有一个类别就用这个

2. 有多种类别

同样修改detect.py文件,添加下面画红框的,就可以了,包括想修改框粗细的,都在下面哦~懒得复制了,大家需要的直接手敲吧。

目标检测YOLOV7 添加计数功能_第2张图片

目标检测YOLOV7 添加计数功能_第3张图片

目标检测YOLOV7 添加计数功能_第4张图片

3. 结果

目标检测YOLOV7 添加计数功能_第5张图片

你可能感兴趣的:(YOLO)