From 90299fc87a17d934995d143bfbc89c07e897aa90 Mon Sep 17 00:00:00 2001 From: philipqueen Date: Wed, 20 Mar 2024 15:56:04 -0600 Subject: [PATCH] Add yolo model size to mediapipe tracking params for yolo crop --- .../trackers/mediapipe_tracker/mediapipe_model_info.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/skellytracker/trackers/mediapipe_tracker/mediapipe_model_info.py b/skellytracker/trackers/mediapipe_tracker/mediapipe_model_info.py index 10768a7..2dc0320 100644 --- a/skellytracker/trackers/mediapipe_tracker/mediapipe_model_info.py +++ b/skellytracker/trackers/mediapipe_tracker/mediapipe_model_info.py @@ -36,6 +36,9 @@ class MediapipeTrackingParams(BaseTrackingParams): min_detection_confidence: float = 0.5 min_tracking_confidence: float = 0.5 static_image_mode: bool = False + yolo_model_size: Literal[ + "nano", "small", "medium", "large", "extra_large", "high_res" + ] = "nano" bounding_box_buffer_percentage: float = 10 buffer_size_method: Literal["buffer_by_box_size", "buffer_by_image_size"] = ( "buffer_by_box_size"