YOLOv5实现目标计数

Albert_yeager

已于 2023-03-13 21:44:07 修改

阅读量1.2w

点赞数 34

分类专栏：无人机视觉文章标签：视觉检测目标检测计算机视觉 YOLO

于 2023-03-13 00:26:18 首次发布

本文链接：https://blog.csdn.net/Albert_yeager/article/details/129482887

版权

无人机视觉专栏收录该内容

17 篇文章

订阅专栏

本文主要讲解如何使用YOLOv5实现目标计数。

在detect.py文件中这部分内容替换为下面代码：

# Write results+计数
count = 1
for *xyxy, conf, cls in reversed(det):
    if save_txt:  # Write to file
        xywh = (xyxy2xywh(torch.tensor(xyxy).view(1, 4)) / gn).view(-1).tolist()  # normalized xywh
        line = (cls, *xywh, conf) if opt.save_conf else (cls, *xywh)  # label format
        with open(txt_path + '.txt', 'a') as f:
            f.write(('%g ' * len(line)).rstrip() % line + '\n')

    if save_img or view_img:  # Add bbox to image
        c = int(cls)# integer class分类数
        label = '%s %.2f  num: %d' % (names[int(cls)], conf, count)
        plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=3)
        count += 1

原理比较简单，就是计算锚框数量，每打印一个框count计数+1（但是值得一提的是，这种方法是不区分类别的，后续我想办法按照类进行计数）。其中label变量记录需要展示的变量，原来是“标签名称+置信度”，修改之后展示的是“标签名称+置信度+计数”。如果还需要展示其它变量继续修改label即可。

效果还行

yolov5-6.x版本和5.0略有差异，但原理相同，只要在detect.py文件进行以上修改即可在标签label中显示计数。加了“！！！”的地方就是要改的地方。

# Write results
count = 1 # ！！！初始计数count=1
for *xyxy, conf, cls in reversed(det):
#...
#这里省略若干行代码
#...
    if save_img or save_crop or view_img:  
        c = int(cls)  # 分类数
        label = None if hide_labels else (names[c] if hide_conf else f'{names[c]} {conf:.2f} {count}') # TODO 标签展示这里加了末尾的{count}
        annotator.box_label(xyxy, label, color=colors(c, True))
        count += 1 # ！！！ 这里加了循环累加

求学路上，你我共勉(๑•̀ㅂ•́)و✧