Update track.py

track.py

@@ -72,7 +72,8 @@ def run(
         augment=False,  # augmented inference
         visualize=False,  # visualize features
         update=False,  # update all models
-        project=ROOT / 'runs' / 'track',  # save results to project/name
+        #project=ROOT / 'runs' / 'track',  # save results to project/name
+        project=ROOT # save results to project/name
         name='exp',  # save results to project/name
         exist_ok=True,  # existing project/name ok, do not increment
         line_thickness=2,  # bounding box thickness (pixels)
@@ -343,9 +344,9 @@ def parse_opt():
     parser.add_argument('--augment', action='store_true', help='augmented inference')
     parser.add_argument('--visualize', action='store_true', help='visualize features')
     parser.add_argument('--update', action='store_true', help='update all models')
-    parser.add_argument('--project', default=ROOT / 'runs' / 'track', help='save results to project/name')
-    parser.add_argument('--name', default='exp', help='save results to project/name')
-    parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment')
+    parser.add_argument('--project', default=ROOT , help='save results to project/name')
+    parser.add_argument('--name', default='exp', help='save results to ROOT')
+    parser.add_argument('--exist-ok', default='True', action='store_true', help='existing project/name ok, do not increment')
     parser.add_argument('--line-thickness', default=2, type=int, help='bounding box thickness (pixels)')
     parser.add_argument('--hide-labels', default=False, action='store_true', help='hide labels')
     parser.add_argument('--hide-conf', default=False, action='store_true', help='hide confidences')
@@ -381,7 +382,7 @@ def MOT(yoloweights, trackingmethod, sourceVideo):
     opt.tracking_config = ROOT / 'trackers' / opt.tracking_method / 'configs' / (opt.tracking_method + '.yaml')
     print_args(vars(opt))
     main(opt)
-    save_dir = increment_path('runs/track/exp', exist_ok=True) 
+    save_dir = increment_path('exp', exist_ok=True) 
     input = os.path.join(save_dir,'out.mp4')
     outpath = 'output.mp4'  #'output/'+ 'output.mp4' 
     command = f"ffmpeg -i {input} -vf fps=30 -vcodec libx264 {outpath}"