
convLSTM模型应用于movingMNIST数据集。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
convLSTM_movingMNIST模型利用convLSTM单元,对移动的MNIST数据集进行帧预测。该项目包含两种类型的模型:一种为单步模型,其结构采用编码器;另一种为多步模型,其结构为编码器-解码器。单步模型接收19个移动的MNIST帧以及一系列的“attemts”作为输入,旨在预测第20帧。多步模型则接收10到19个移动MNIST帧(每个批次随机选择的序列长度),并尝试预测剩余帧的序列。在训练阶段,单步模型的权重将被用于初始化编码器部分,从而加速模型的收敛速度(特别是在从头开始训练时)。请注意,运行此项目需要Tensorflow v1.4或更高版本,并且如果希望训练多步模型,可能需要配备GPU资源。 为了获取数据集,可以通过以下命令进行克隆:git clone https://github.com/Linusnie/convLSTM_movingMNIST.git,然后进入存储库目录:cd convLSTM_movingMNIST。最后,下载移动的mnist数据集:wget http://www.cs.toronto.edu/~niti
全部评论 (0)
还没有任何评论哟~


