MCG-NJU
/

videomae-base

Video Classification

Model card Files Files and versions

nielsr HF Staff commited on Apr 22, 2023

Commit

ef591f7

·

1 Parent(s): 0dea0db

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -38,7 +38,7 @@ video = list(np.random.randn(16, 3, 224, 224))
 processor = VideoMAEImageProcessor.from_pretrained("MCG-NJU/videomae-base")
 model = VideoMAEForPreTraining.from_pretrained("MCG-NJU/videomae-base")
-pixel_values = feature_extractor(video, return_tensors="pt").pixel_values
 num_patches_per_frame = (model.config.image_size // model.config.patch_size) ** 2
 seq_length = (num_frames // model.config.tubelet_size) * num_patches_per_frame

 processor = VideoMAEImageProcessor.from_pretrained("MCG-NJU/videomae-base")
 model = VideoMAEForPreTraining.from_pretrained("MCG-NJU/videomae-base")
+pixel_values = processor(video, return_tensors="pt").pixel_values
 num_patches_per_frame = (model.config.image_size // model.config.patch_size) ** 2
 seq_length = (num_frames // model.config.tubelet_size) * num_patches_per_frame