当前位置:   article > 正文

YOLOv7怎么改变检测框(box)的标签字体样式与大小,检测框颜色,修改字体宽度;固定检测标签颜色_yolov7的框

yolov7的框

YOLOv7detect.py所用的检测框字体为plots.py中的 plot_one_box函数

from utils.plots import plot_one_box

因此进入到utils文件夹下的plots.py文件中,找到plot_one_box

  1. def plot_one_box(x, img, color=None, label=None, line_thickness=3):
  2. # Plots one bounding box on image img
  3. tl = line_thickness or round(0.002 * (img.shape[0] + img.shape[1]) / 2) + 1 # line/font thickness
  4. color = color or [random.randint(0, 255) for _ in range(3)]
  5. c1, c2 = (int(x[0]), int(x[1])), (int(x[2]), int(x[3]))
  6. cv2.rectangle(img, c1, c2, color, thickness=tl, lineType=cv2.LINE_AA)
  7. if label:
  8. tf = max(tl - 1, 1) # font thickness
  9. t_size = cv2.getTextSize(label, 0, fontScale=tf/3 , thickness=tf)[0]
  10. c2 = c1[0] + t_size[0], c1[1] - t_size[1] - 3
  11. cv2.rectangle(img, c1, c2, color, -1, cv2.LINE_AA) # filled
  12. cv2.putText(img, label, (c1[0], c1[1] - 2), 0, 2, [225, 255, 255], thickness=tf, lineType=cv2.LINE_AA)

1. 修改标签背景框:

找到t_size = cv2.getTextSize(label, 0, fontScale=tl/3 , thickness=tl)[0],对这一行进行修改:

      fontscale=1.5                                                        fontscale=3

                                        

 thickness我改后发现没什么变化

2. 修改标签字体:

找到 cv2.putText(img, label, (c1[0], c1[1] - 2), 0, 2, [225, 255, 255], thickness=tf, lineType=cv2.LINE_AA),对这一行进行修改:

括号里的参数依次为:图片,添加的文字,左上角坐标,字体样式字体大小颜色字体粗细

因此,想要修改字体大小,需要对第3个参数进行修改,

cv2.putText(img, label, (c1[0], c1[1] - 2), 0, 3, [225, 255, 255], thickness=tf, lineType=cv2.LINE_AA)

一般来讲,字体大小数值和标签背景框的大小相差不大。

3. 修改字体宽度:

一般你的标签看不清,都需要改这一项,找到

cv2.putText(img, label, (c1[0], c1[1] - 2), 0, 2, [225, 255, 255], thickness=tf, lineType=cv2.LINE_AA)

可以直接tf改成你想要的数字,或者改tf参数

cv2.putText(img, label, (c1[0], c1[1] - 2), 0, 2, [225, 255, 255], thickness=2, lineType=cv2.LINE_AA)

4. 修改字体样式:

cv2.putText(img, label, (c1[0], c1[1] - 2), 0, 2, [225, 255, 255], thickness=tf, lineType=cv2.LINE_AA)

5. 修改框的颜色:

因为YOLOv7颜色框每run一次exp就会随机分配一次,因此为了做不同网络对比,需要对每次run的检测框颜色进行定义。

修改颜色直接进detect.py文件中找到:

  1. names = model.module.names if hasattr(model, 'module') else model.names
  2. colors = [[random.randint(0, 255) for _ in range(3)] for _ in names]

可见每次detect后颜色都是随机生成的,因此我们需要把每类标签的颜色固定。

把colors改为你想要的颜色,有多少个类别就选择多少颜色,也可以多列举几个颜色 :

  1. colors = [[0, 255, 0], [0, 0, 255], [255, 0, 0],[255, 150, 0],
  2. [255, 0, 150], [0, 150, 255], [150, 0, 255]]

本人也是近期接触YOLOv7,如果有什么更好的修改方法欢迎评论区互相讨论学习! 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小丑西瓜9/article/detail/386095
推荐阅读
相关标签
  

闽ICP备14008679号