thuml/predrnn-pytorch

Generate a new Moving-MNIST dataset

Mareeta26 opened this issue 2 years ago · 1 comments

Mareeta26 commented 2 years ago

Hi,
May I know where can I find the code to generate a new Moving-MNIST dataset? I need to create subsets of different sizes.
Thank you,
Mareeta

wuhaixu2016 commented 2 years ago

Hi, you may find this useful.
https://gitlab.fbk.eu/deep-learning/meteo-tn/-/blob/0beb24ead64ee37315b54eba1c7d70e384145ca6/nowcasting/movingmnist_iterator.py

👍1