nielsr HF staff commited on
Commit
ef591f7
1 Parent(s): 0dea0db

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -38,7 +38,7 @@ video = list(np.random.randn(16, 3, 224, 224))
38
  processor = VideoMAEImageProcessor.from_pretrained("MCG-NJU/videomae-base")
39
  model = VideoMAEForPreTraining.from_pretrained("MCG-NJU/videomae-base")
40
 
41
- pixel_values = feature_extractor(video, return_tensors="pt").pixel_values
42
 
43
  num_patches_per_frame = (model.config.image_size // model.config.patch_size) ** 2
44
  seq_length = (num_frames // model.config.tubelet_size) * num_patches_per_frame
 
38
  processor = VideoMAEImageProcessor.from_pretrained("MCG-NJU/videomae-base")
39
  model = VideoMAEForPreTraining.from_pretrained("MCG-NJU/videomae-base")
40
 
41
+ pixel_values = processor(video, return_tensors="pt").pixel_values
42
 
43
  num_patches_per_frame = (model.config.image_size // model.config.patch_size) ** 2
44
  seq_length = (num_frames // model.config.tubelet_size) * num_patches_per_frame