facebook
/

timesformer-hr-finetuned-k600

Video Classification

Inference Endpoints

Model card Files Files and versions Community

fcakyon commited on Dec 2, 2022

Commit

283d3da

•

1 Parent(s): d4a0916

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -24,7 +24,7 @@ from transformers import VideoMAEFeatureExtractor, TimesformerForVideoClassifica
 import numpy as np
 import torch
-video = list(np.random.randn(8, 3, 224, 224))
 feature_extractor = VideoMAEFeatureExtractor.from_pretrained("MCG-NJU/videomae-base-finetuned-kinetics")
 model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-hr-finetuned-k600")

 import numpy as np
 import torch
+video = list(np.random.randn(16, 3, 448, 448))
 feature_extractor = VideoMAEFeatureExtractor.from_pretrained("MCG-NJU/videomae-base-finetuned-kinetics")
 model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-hr-finetuned-k600")