huxian0402 / ConvLSTM-Moving-mnist

Using convlstm to prediction moving mnist dataset.
MIT License
23 stars 7 forks source link

ConvLSTM-Moving-mnist

Using convlstm to prediction moving mnist dataset.

run main.py for train

1、network structure 使用其中一个序列,迭代训练,收敛。 前10帧--第11帧,...,第10-19帧预测第20帧。 输入mnist序列 三层conv 一层BasicConvLstmCel 三层conv flatten (1,10,64,64,1) (1,10,64,64,16)(1,1,64,64,1024) (1,1,64,64,1)

2、result image image

3、loss curve image