Spaces:
Runtime error
Runtime error
Commit
·
40b826b
1
Parent(s):
f5c50b7
fixing args issue
Browse files
track.py
CHANGED
@@ -320,11 +320,11 @@ def run(
|
|
320 |
|
321 |
def parse_opt():
|
322 |
parser = argparse.ArgumentParser()
|
323 |
-
parser.add_argument('--yolo-weights', nargs='+', type=Path, default=WEIGHTS / 'yolov8s-seg.pt', help='model.pt path(s)')
|
324 |
parser.add_argument('--reid-weights', type=Path, default=WEIGHTS / 'osnet_x0_25_msmt17.pt')
|
325 |
-
parser.add_argument('--tracking-method', type=str, default='bytetrack', help='strongsort, ocsort, bytetrack')
|
326 |
parser.add_argument('--tracking-config', type=Path, default=None)
|
327 |
-
parser.add_argument('--source', type=str, default='0', help='file/dir/URL/glob, 0 for webcam')
|
328 |
parser.add_argument('--imgsz', '--img', '--img-size', nargs='+', type=int, default=[640], help='inference size h,w')
|
329 |
parser.add_argument('--conf-thres', type=float, default=0.5, help='confidence threshold')
|
330 |
parser.add_argument('--iou-thres', type=float, default=0.5, help='NMS IoU threshold')
|
@@ -375,7 +375,7 @@ def MOT(yoloweights, trackingmethod, sourceVideo):
|
|
375 |
parser = parse_opt()
|
376 |
parser.add_argument('--yolo-weights', nargs='+', type=Path, default= yoloweights, help='model.pt path(s)')
|
377 |
parser.add_argument('--tracking-method', type=str, default= trackingmethod, help='strongsort, ocsort, bytetrack')
|
378 |
-
parser.add_argument('--source', type=str, default=
|
379 |
opt = parser.parse_args()
|
380 |
opt.imgsz *= 2 if len(opt.imgsz) == 1 else 1 # expand
|
381 |
opt.tracking_config = ROOT / 'trackers' / opt.tracking_method / 'configs' / (opt.tracking_method + '.yaml')
|
|
|
320 |
|
321 |
def parse_opt():
|
322 |
parser = argparse.ArgumentParser()
|
323 |
+
#parser.add_argument('--yolo-weights', nargs='+', type=Path, default=WEIGHTS / 'yolov8s-seg.pt', help='model.pt path(s)')
|
324 |
parser.add_argument('--reid-weights', type=Path, default=WEIGHTS / 'osnet_x0_25_msmt17.pt')
|
325 |
+
#parser.add_argument('--tracking-method', type=str, default='bytetrack', help='strongsort, ocsort, bytetrack')
|
326 |
parser.add_argument('--tracking-config', type=Path, default=None)
|
327 |
+
#parser.add_argument('--source', type=str, default='0', help='file/dir/URL/glob, 0 for webcam')
|
328 |
parser.add_argument('--imgsz', '--img', '--img-size', nargs='+', type=int, default=[640], help='inference size h,w')
|
329 |
parser.add_argument('--conf-thres', type=float, default=0.5, help='confidence threshold')
|
330 |
parser.add_argument('--iou-thres', type=float, default=0.5, help='NMS IoU threshold')
|
|
|
375 |
parser = parse_opt()
|
376 |
parser.add_argument('--yolo-weights', nargs='+', type=Path, default= yoloweights, help='model.pt path(s)')
|
377 |
parser.add_argument('--tracking-method', type=str, default= trackingmethod, help='strongsort, ocsort, bytetrack')
|
378 |
+
parser.add_argument('--source', type=str, default=sourceVideo, help='file/dir/URL/glob, 0 for webcam')
|
379 |
opt = parser.parse_args()
|
380 |
opt.imgsz *= 2 if len(opt.imgsz) == 1 else 1 # expand
|
381 |
opt.tracking_config = ROOT / 'trackers' / opt.tracking_method / 'configs' / (opt.tracking_method + '.yaml')
|