facebook
/
timesformer-base-finetuned-k400
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-12-06 09:56:23 +00:00 committed by huggingface-web
parent b5dfd1b518
commit 45770ed498
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -26,7 +26,7 @@ import torch
video = list(np.random.randn(8, 3, 224, 224)) video = list(np.random.randn(8, 3, 224, 224))
feature_extractor = AutoImageProcessor.from_pretrained("MCG-NJU/videomae-base-finetuned-kinetics") feature_extractor = AutoImageProcessor.from_pretrained("facebook/timesformer-base-finetuned-k400")
model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-base-finetuned-k400") model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-base-finetuned-k400")
inputs = feature_extractor(video, return_tensors="pt") inputs = feature_extractor(video, return_tensors="pt")