Summary of YOLOv5 unpopular knowledge

introduction

Here is a summary of some small changes that are more partial, and will continue to be updated. Although these do not seem to be very useful, but in order to be able to call directly when needed, I still want to summarize them for easy reference.

Recognition box line thickness modification

Change the tl value in the plot_one_box function of plots.py

tl = 30

default

Line thickness is 10

Line thickness is 30

Recognition box color modification

color = (255,0,255)

(255,0,255) represents purple

The current changes to the plot_one_box() function are as follows

def plot_one_box(x, img, color=None, label=None, line_thickness=3):
    # 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
    #tl = 10
    color = color or [random.randint(0, 255) for _ in range(3)]
    #color = (255,0,255)
    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)
    # #################################打印坐标#############################
    # print("左上点的坐标为：(" + str(c1[0]) + "," + str(c1[1]) + ")，右上点的坐标为(" + str(c2[0]) + "," + str(c1[1]) + ")")
    # print("左下点的坐标为：(" + str(c1[0]) + "," + str(c2[1]) + ")，右下点的坐标为(" + str(c2[0]) + "," + str(c2[1]) + ")")
    # print("中心点的坐标为：(" + str((c2[0] - c1[0]) / 2 + c1[0]) + "," + str((c2[1] - c1[1]) / 2 + c1[1]) + ")")
    # #####################################################################
    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)

label display modification

label = '%s %.2f  xxx: %d' % (names[int(cls)], conf, 1234) 
#改成标准输出格式，方便添加其他变量（原代码只显示类别名和置信度）

before fixing

After modification (here is just an example, no special meaning)

On the way to school, you and I encourage each other (๑•̀ㅂ•́)و✧