mx-mark / VideoTransformer-pytorch

PyTorch implementation of a collections of scalable Video Transformer Benchmarks.
272 stars 34 forks source link

How Can I Create a Video_Loader Function That Lets Me Use My Own Videos With ViViT? #35

Open coolbagel opened 1 year ago

coolbagel commented 1 year ago

As the title says, I would like to create a function that prepares a folder of videos for use with the ViViT transformer, but am having trouble with where to start. Any point in the right direction would be incredibly helpful.