
ConvLSTM-Moving-mnist:利用ConvLSTM模型预测运动mnist数据集。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
通过利用ConvLSTM网络对移动MNIST数据集进行预测,我们运行了主程序`main.py`以进行训练。该网络结构采用其中一个序列进行迭代训练,直至收敛。在最初的前10帧以及后续的第11帧至第19帧之间,网络负责预测第20帧的结果。具体而言,输入的是MNIST序列,该序列经过三层卷积层和一层BasicConvLstmCell层处理,随后进行flatten操作,最终得到(1, 10, 64, 64, 1)、(1, 10, 64, 64, 16)和(1, 1, 64, 64, 1024)三个特征图,最后再通过一层(1, 1, 64, 64, 1)的卷积层完成。
全部评论 (0)
还没有任何评论哟~


