Unsupervised learning of Moving MNIST dataset. This repository contains implemention of ConvLSTM model (https://arxiv.org/abs/1506.04214) and PredRNN++(https://arxiv.org/abs/1804.06300) model with Pytorch. The ConvLSTM model is based on https://github.com/jhhuang96/ConvLSTM-PyTorch. The PredRNN++ model is based on https://github.com/thuml/predrnn-pytorch.
The dataset can be downloaded from https://www.cs.toronto.edu/~nitish/unsupervised_video/
对移动数字数据集的无监督学习模型。本项目包括基于Pytorch实现的ConvLSTM模型(https://arxiv.org/abs/1506.04214)和PredRNN++模型(https://arxiv.org/abs/1804.06300)。 ConvLSTM模型实现基于https://github.com/jhhuang96/ConvLSTM-PyTorch PredRNN++模型实现基于https://github.com/thuml/predrnn-pytorch
本数据集下载地址：https://www.cs.toronto.edu/~nitish/unsupervised_video/