-
使用ConvLSTM对移动的mnist数据集进行预测:ConvLSTM-Moving-mnist
资源介绍
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
3、loss curve