开学我就研二下了,感叹时光飞逝。因为好多人在评论里问,所以出一篇!注:本文章应该是适用所有YOLO系列。
1. 只有一种类别
修改detect.py文件,找到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) # 只有一个类别就用这个
2. 有多种类别
同样修改detect.py文件,添加下面画红框的,就可以了,包括想修改框粗细的,都在下面哦~懒得复制了,大家需要的直接手敲吧。