From 591c5b0236c127da1bf949975a4f8bb79c6cd156 Mon Sep 17 00:00:00 2001 From: AlexeyAB84 Date: Sat, 6 Aug 2022 01:45:25 +0300 Subject: [PATCH] minor fix --- detect.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/detect.py b/detect.py index f6efbea..5648a91 100644 --- a/detect.py +++ b/detect.py @@ -125,10 +125,10 @@ def detect(save_img=False): if save_img or view_img: # Add bbox to image label = f'{names[int(cls)]} {conf:.2f}' - plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=3) + plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=1) # Print time (inference + NMS) - #print(f'{s}Done. ({(1E3 * (t2 - t1)):.1f}ms) Inference, ({(1E3 * (t3 - t2)):.1f}s) NMS') + print(f'{s}Done. ({(1E3 * (t2 - t1)):.1f}ms) Inference, ({(1E3 * (t3 - t2)):.1f}ms) NMS') # Stream results if view_img: