fix: add labels parameter to ActionRecognition class constructor

5 months ago · e4ed047699
parent d72e23a18f
commit e4ed047699
1 changed files with 1 additions and 0 deletions
--- a/ultralytics/solutions/action_recognition.py
+++ b/ultralytics/solutions/action_recognition.py
@ -20,6 +20,7 @@ class ActionRecognition:
    def __init__(
        self,
        video_classifier_model="microsoft/xclip-base-patch32",
+        labels: List[str] = None,
        crop_margin_percentage: int = 10,
        num_video_sequence_samples: int = 8,
        vid_stride: int = 2,