HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Fix Action Recognition Example with torch>=2.0 (#14232)

Browse source
This commit is contained in:
fatih c. akyon 2024-07-05 20:25:42 +03:00 • committed by GitHub
parent 5d479c73c2
commit 2b1b26333b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
examples/YOLOv8-Action-Recognition/action_recognition.py
View file

@ -171,13 +171,13 @@ class HuggingFaceVideoClassifier:
""" """
if input_size is None: if input_size is None:
input_size = [224, 224] input_size = [224, 224]
from torchvision.transforms import v2 from torchvision import transforms
transform = v2.Compose( transform = transforms.Compose(
[ [
v2.ToDtype(torch.float32, scale=True), transforms.Lambda(lambda x: x.float() / 255.0),
v2.Resize(input_size, antialias=True), transforms.Resize(input_size),
v2.Normalize( transforms.Normalize(
mean=self.processor.image_processor.image_mean, std=self.processor.image_processor.image_std mean=self.processor.image_processor.image_mean, std=self.processor.image_processor.image_std
), ),
] ]