YOLOv5-2.0如何更改标签颜色和字体大小

在yolov5-2.0项目文件下找到utils目录，打开utils.py文件，找到“plot_one_box()”函数，大概是在第940行位置。

内容如下：

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

1、修改标签背景框，即找到下面代码，修改参数“ fontScale”，如“ fontScale=3=5”。

t_size = cv2.getTextSize(label, 0, fontScale=tl / 3, thickness=tf)[0]

2、修改标签字体格式，即找到下面代码 进行修改，参数“img, label, (c1[0], c1[1] - 2), 0, tl / 3, [225, 255, 255], thickness=tf”依次代表“图片，添加的文字，左上角坐标，字体样式，字体大小，颜色，字体粗细”，参数的值可以是0、1、2、3、4、5等等。

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

字体样式如下：

3、取消检测标签，只保留检测框,注释掉若干行代码即可。

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

4、修改随机颜色，可见每次detect后颜色都是随机生成的，因此我们需要把每类标签的颜色固定。把colors改为你想要的颜色，有多少个类别就选择多少颜色，也可以多列举几个颜色 ：

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

修改后：

colors = [[0, 255, 0], [0, 0, 255], [255, 0, 0],[255, 150, 0],
    [255, 0, 150], [0, 150, 255], [150, 0, 255]]