Merge pull request #30 from freemocap/philip/get_yolo_crop_tracking_p…

…arams Patch: pass yolo crop tracking params into process folder of videos
freemocap · Mar 26, 2024 · 1ecb538 · 1ecb538
2 parents aaaed17 + 7145dff
commit 1ecb538
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/skellytracker/process_folder_of_videos.py b/skellytracker/process_folder_of_videos.py
@@ -148,10 +148,13 @@ def get_tracker(tracker_name: str, tracking_params: BaseModel) -> BaseTracker:
 
     elif tracker_name == "YOLOMediapipeComboTracker":
         tracker = YOLOMediapipeComboTracker(
+            model_size=tracking_params.yolo_model_size,
             model_complexity=tracking_params.mediapipe_model_complexity,
             min_detection_confidence=tracking_params.min_detection_confidence,
             min_tracking_confidence=tracking_params.min_tracking_confidence,
             static_image_mode=True,  # yolo cropping must be run with static image mode due to changing size of bounding boxes
+            bounding_box_buffer_percentage=tracking_params.bounding_box_buffer_percentage,
+            buffer_size_method=tracking_params.buffer_size_method,
         )
 
     elif tracker_name == "YOLOPoseTracker":