MCG-NJU
/
videomae-base-finetuned-kinetics
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md

This commit is contained in:
Niels Rogge 2022-08-03 10:17:01 +00:00 committed by huggingface-web
parent 422e706f9b
commit 4800870825
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

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