XiYe20 / VPTR

The repository for paper VPTR: Efficient Transformers for Video Prediction
MIT License
88 stars 19 forks source link

Can you post the instructions to do inference from custom frames too? #12

Closed mayank64ce closed 1 year ago

XiYe20 commented 1 year ago

Hi, you can use custom frames by replacing the "past_frames" in Test_VPTR.ipynb with your own custom frames. But please remember to resize and normalize your custom frames as what utils/dataset.py does. Please see the "get_dataloader()" function for all transforms applied to the datasets.