MCG-NJU
/

videomae-base-finetuned-kinetics

Video Classification

Inference Endpoints

Model card Files Files and versions Community

nielsr HF staff commited on Aug 3, 2022

Commit

4800870

•

1 Parent(s): 422e706

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -34,8 +34,8 @@ import torch
 video = list(np.random.randn(16, 3, 224, 224))
-feature_extractor = VideoMAEFeatureExtractor.from_pretrained("MCG-NJU/videomae-base")
-model = VideoMAEForVideoClassification.from_pretrained("MCG-NJU/videomae-base")
 inputs = feature_extractor(video, return_tensors="pt")

 video = list(np.random.randn(16, 3, 224, 224))
+feature_extractor = VideoMAEFeatureExtractor.from_pretrained("MCG-NJU/videomae-base-finetuned-kinetics")
+model = VideoMAEForVideoClassification.from_pretrained("MCG-NJU/videomae-base-finetuned-kinetics")
 inputs = feature_extractor(video, return_tensors="pt")