YOLOV5/V7 实现图片、视频、摄像头标签数量统计

BABABSBC

已于 2023-03-12 01:27:49 修改

阅读量769

点赞数 1

文章标签：目标检测 YOLO python 深度学习 Powered by 金山文档

于 2023-03-12 00:41:37 首次发布

本文链接：https://blog.csdn.net/BABABSBC/article/details/129471189

版权

本文主要是在推理detect阶段，预测的图片或视频中没有标签的数量统计

在 detect.py中的第171行（大家看自己位置，我这里是大概位置）加入这行代码

cv2.putText(im0, f"{n} " + 'person', (5, 50), cv2.FONT_HERSHEY_SIMPLEX, 1.2, (0, 0, 255), 2)

修改后的 11行和28行29行

            if len(det):
                # Rescale boxes from img_size to im0 size
                det[:, :4] = scale_coords(im.shape[2:], det[:, :4], im0.shape).round()

                # Print results
                for c in det[:, -1].unique():
                    n = (det[:, -1] == c).sum()  # detections per class
                    print(n)
                    s += f"{n} {names[int(c)]}{'s' * (n > 1)}, "  # add to string

                    cv2.putText(im0, f"{n} " +