How to make output bounding boxes thinner ?

[Samuel-Bachorik]

Hi ! Now for me Yolo is working perfectly but I have one question about output boxes. When I use low resolution of video these boxes are very Thick. I want them to be thinner on low resolution images. Can you please show me what should I adjust ?
Photo what I mean - https://ibb.co/bFMSXXS Text and boxes are very Thick and big

Thank you

[kinoute]

You can try to change the line_thickness parameter of the plot_one_box() function here:

yolov5/utils/plots.py

Line 57 in 886f1c0

def plot_one_box(x, img, color=None, label=None, line_thickness=3):

The default is set to 3, 1 might be better.

[glenn-jocher]

We probably want a better linewidth and fontsize scaling method that works across a wider range of sizes (i.e. 128-4k). We attempt some autoscaling now (L59, L65) but it mostly looks best at 640-1280.

yolov5/utils/plots.py

Lines 57 to 69 in 886f1c0

	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
	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)