run botsort file in src dir to see sample, check in with eric and mic…

…haal to see if its too slow
CornellDataScience · Sep 26, 2023 · 4fd2421 · 4fd2421
1 parent 19a7a05
commit 4fd2421
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/src/botsort.py b/src/botsort.py
@@ -5,7 +5,7 @@
 model = YOLO('yolov8n.pt')
 
 # Open the video file
-video_path = "test2_10seconds.mp4"
+video_path = "training_data.mp4"
 cap = cv2.VideoCapture(video_path)
 
 # Loop through the video frames
@@ -16,7 +16,7 @@
     if success:
         # Run YOLOv8 tracking on the frame, persisting tracks between frames
         # change tracker to botsort.yaml or bytetrack.yaml
-        results = model.track(frame, persist=True, tracker="bytetrack.yaml")
+        results = model.track(frame, persist=True, tracker="botsort.yaml")
 
         # Visualize the results on the frame
         annotated_frame = results[0].plot()

diff --git a/src/training_data.mp4 b/src/training_data.mp4
diff --git a/src/yolov8n.pt b/src/yolov8n.pt