rachana219 commited on
Commit
40b826b
·
1 Parent(s): f5c50b7

fixing args issue

Browse files
Files changed (1) hide show
  1. track.py +4 -4
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='0', 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')
 
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')