fix imgsz error (#41)

* fix imgsz error * update tov 5.0.10
fcakyon · Sep 17, 2021 · 0d03de6 · 0d03de6
1 parent 4aaf354
commit 0d03de6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/yolov5/__init__.py b/yolov5/__init__.py
@@ -1,4 +1,4 @@
 from yolov5.helpers import YOLOv5
 from yolov5.helpers import load_model as load
 
-__version__ = "5.0.9"
+__version__ = "5.0.10"
diff --git a/yolov5/detect.py b/yolov5/detect.py
@@ -53,6 +53,9 @@ def run(weights='yolov5s.pt',  # model.pt path(s)
     save_img = not nosave and not source.endswith('.txt')  # save inference images
     webcam = source.isnumeric() or source.endswith('.txt') or source.lower().startswith(
         ('rtsp://', 'rtmp://', 'http://', 'https://'))
+
+    if isinstance(imgsz, int):
+        imgsz = [imgsz, imgsz]
 
     # Directories
     save_dir = increment_path(Path(project) / name, exist_ok=exist_ok)  # increment run