From c6380d14a0022ffdcddfb9b0d6cd1e316c7d9329 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mikel=20Brostr=C3=B6m?= Date: Fri, 20 Dec 2024 09:39:16 +0100 Subject: [PATCH] fix when no model selected --- tracking/val.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tracking/val.py b/tracking/val.py index 3efa24ffa..263f5ba26 100644 --- 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')