XiYe20 / VPTR

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

how to convert official BAIR dataset #10

Closed JunyaoHu closed 1 year ago

JunyaoHu commented 1 year ago

Hello dear author, I want to know how to convert official BAIR dataset to your dataset format. Thanks!

/BAIR
     test/
         example_0/
            0000.png
            0001.png
XiYe20 commented 1 year ago

Hi, please see: utils/read_BAIR_tfrecords.py

JunyaoHu commented 1 year ago

Thank you. It was an oversight of mine.