Update tracker.py

now processes the training video
CornellDataScience · Sep 17, 2023 · 3a6e5c2 · 3a6e5c2
1 parent 8934cb6
commit 3a6e5c2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/StrongSORT-YOLO/tracker.py b/src/StrongSORT-YOLO/tracker.py
@@ -28,8 +28,9 @@ def test():
     print('import worked')
 
 if __name__ == '__main__':
-    output = get_data('../../data/benson.mp4')
+    # TODO change to actual video path
+    output = get_data('../../data/training_data.mp4')
     print(output['basketball_data'])
     with open('../../tmp/test_output.pickle', 'wb') as f:
         pickle.dump(output, f)
-
+