目标检测YOLOV5 添加计数功能

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

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

原来的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)
 
 

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

 

猜你喜欢

转载自blog.csdn.net/doumaidexiaozhi/article/details/127519723