Initial commit

track.py

@@ -59,7 +59,7 @@ def run(
         save_conf=False,  # save confidences in --save-txt labels
         save_crop=False,  # save cropped prediction boxes
         save_trajectories=False,  # save trajectories for each track
-        save_vid=False,  # save confidences in --save-txt labels
+        save_vid=True,  # save confidences in --save-txt labels
         nosave=False,  # do not save images/videos
         classes=None,  # filter by class: --class 0, or --class 0 2 3
         agnostic_nms=False,  # class-agnostic NMS
@@ -68,7 +68,7 @@ def run(
         update=False,  # update all models
         project=ROOT / 'runs' / 'track',  # save results to project/name
         name='exp',  # save results to project/name
-        exist_ok=False,  # existing project/name ok, do not increment
+        exist_ok=True,  # existing project/name ok, do not increment
         line_thickness=2,  # bounding box thickness (pixels)
         hide_labels=False,  # hide labels
         hide_conf=False,  # hide confidences
@@ -367,4 +367,10 @@ def main(opt):
 def MOT():
     opt = parse_opt()
     main(opt)
+    save_dir = increment_path('runs/track/exp', exist_ok=False) 
+    input = os.path.join(save_dir,'out.mp4')
+    tracking_method='strongsort'
+    outpath = 'output/' + tracking_method + '_output.mp4' 
+    #!ffmpeg -i $input -vf fps=30 -vcodec libx264 $outpath tbd
+