action-recog
fcakyon 4 months ago
parent dc061fe770
commit ce0404547c
  1. 2
      ultralytics/solutions/action_recognition.py

@ -378,7 +378,7 @@ class HuggingFaceVideoClassifier:
if input_size is None: if input_size is None:
input_size = [224, 224] input_size = [224, 224]
if self.upports_transforms_v2: if self.supports_transforms_v2:
from torchvision.transforms import v2 from torchvision.transforms import v2
transform = v2.Compose( transform = v2.Compose(

Loading…
Cancel
Save