目标检测YOLOV5 添加计数功能

YOLOV5预测完图片想显示个数怎么办呢?

一行代码轻松解决!!!!

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

原来的Detect 没有计数功能

只需在源码

if save_img:

上加上下面这一段代码即可:

cv2.putText(im0, f"{n} {names[int(c)]}{'s' * (n > 1)}", (5, 50), cv2.FONT_HERSHEY_SIMPLEX, 1.2,(0, 0, 255), 2)
 
  

这样就可以加上计数功能了!!!!

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

你可能感兴趣的:(YOLOV5应用,目标检测,计算机视觉,深度学习)