Convolutional LSTM implemented with chainer
Switch branches/tags
Nothing to show
Clone or download
Latest commit dc8dcac Oct 27, 2017
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
imgs add images Oct 27, 2017
LICENSE Initial commit Oct 27, 2017
README.md upd README Oct 27, 2017
dataset.py fix dataset path Oct 27, 2017
generate.py first commit Oct 27, 2017
get-moving-mnist.sh first commit Oct 27, 2017
network.py first commit Oct 27, 2017
train.py first commit Oct 27, 2017

README.md

ConvLSTM

Convolutional LSTM implemented with chainer

python 3.5.2 + chainer 3.0.0

Getting dataset

$ ./get-moving-mnist.sh

Training

$ python3 ./train.py -g 0 --epoch 10 --inf 3 --outf 3 --batch 16

Generating

$ mkdir img
$ python3 ./generate.py --model results/model --id 7000 --inf 3 --outf 3

Then, the images are generated in img/.

$ mkdir img
$ wget -O model https://www.dropbox.com/s/pthwoljp8qq3o30/model?dl=0
$ python3 ./generate.py --model model --id 7000 --inf 3 --outf 3

Or, you can try ConvLSTM with the pre-trained model.

Gallery

7000_cap

7001

7002

7003

7004