From 19c56e60b100cf8ff9af65b4347de69e0cff76ae Mon Sep 17 00:00:00 2001 From: Diego Montes <54745152+d57montes@users.noreply.github.com> Date: Sun, 12 Dec 2021 17:39:14 -0500 Subject: [PATCH] Fix `imgsz` bug (#5948) * fix imgsz bug * Update detect.py Co-authored-by: Glenn Jocher --- detect.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/detect.py b/detect.py index 0b6875e55..14cdf96ca 100644 --- a/detect.py +++ b/detect.py @@ -38,7 +38,7 @@ from utils.torch_utils import select_device, time_sync @torch.no_grad() def run(weights=ROOT / 'yolov5s.pt', # model.pt path(s) source=ROOT / 'data/images', # file/dir/URL/glob, 0 for webcam - imgsz=640, # inference size (pixels) + imgsz=(640, 640), # inference size (height, width) conf_thres=0.25, # confidence threshold iou_thres=0.45, # NMS IOU threshold max_det=1000, # maximum detections per image