Merge pull request #1777 from mikel-brostrom/args-as-list

fix when no model selected
mikel-brostrom · Dec 29, 2024 · 90797e3 · 90797e3
2 parents 12e95e8 + c6380d1
commit 90797e3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tracking/val.py b/tracking/val.py
@@ -445,8 +445,8 @@ def parse_opt() -> argparse.Namespace:
     parser = argparse.ArgumentParser()
 
     # Global arguments
-    parser.add_argument('--yolo-model', nargs='+', type=Path, default=WEIGHTS / 'yolov8n.pt', help='yolo model path')
-    parser.add_argument('--reid-model', nargs='+', type=Path, default=WEIGHTS / 'osnet_x0_25_msmt17.pt', help='reid model path')
+    parser.add_argument('--yolo-model', nargs='+', type=Path, default=[WEIGHTS / 'yolov8n.pt'], help='yolo model path')
+    parser.add_argument('--reid-model', nargs='+', type=Path, default=[WEIGHTS / 'osnet_x0_25_msmt17.pt'], help='reid model path')
     parser.add_argument('--source', type=str, help='file/dir/URL/glob, 0 for webcam')
     parser.add_argument('--imgsz', '--img', '--img-size', nargs='+', type=int, default=None, help='inference size h,w')
     parser.add_argument('--conf', type=float, default=0.5, help='confidence threshold')