How to change the dataset

[lihaolin1]

I just want to try moving mnist dataset, is there have any information about change dataset to moving mnist? Thanks!

[jlko]

Dear lihaolin1,

thanks for your message and interest in STOVE! We don't have support for moving MNIST built-in. However, it should be very possible to built it in yourself.

A) If you already have a moving-MNIST dataset.

• I would suggest you generate some of the default moving balls datasets, by just executing run_scripts.py --create-data.
• Have a look at the contents of the pickle files in data/. It should be a pretty straight-forward dictionary that contains the raw data as well as some metadata.
• Export the moving-MNIST data in a pkl file similar to what we expect and replace it with the bouncy balls dataset.

B) If you don't already have a moving-MNIST dataset.

• Have a look at how we create the sprites data in model/envs/envs.py.
• If you replace self.draw_image with something that renders MNIST digits instead of sprites, you should be able to create a moving-MNIST dataset.

Hope this helped, feel free to reach out with further questions. Best Jannik