Fix imgsz bug (#5948)

* fix imgsz bug * Update detect.py Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
SecretStar112 · Dec 12, 2021 · 32f9d1c · 32f9d1c
1 parent 5536f06
commit 32f9d1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/detect.py b/detect.py
@@ -38,7 +38,7 @@
 @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