Update crop_and_pad function signature with type hints

action-recog
fcakyon 4 months ago
parent 887062257f
commit 9f06cbf3b9
  1. 2
      ultralytics/solutions/action_recognition.py

@ -457,7 +457,7 @@ class HuggingFaceVideoClassifier:
return pred_labels, pred_confs
def crop_and_pad(frame, box, margin_percent: int = 10) -> np.ndarray:
def crop_and_pad(frame: np.ndarray, box: List[int], margin_percent: int = 10) -> np.ndarray:
"""
Crop box with margin and take square crop from frame.

Loading…
Cancel
Save