上传者: 42173205
|
上传时间: 2021-12-08 11:38:58
|
文件大小: 48.08MB
|
文件类型: -
概述
自我时空ConvLSTM用于时空预测
SaConvSTLM在此描述的
在MovingMNIST 上MovingMNIST测试
项目结构
load_data.py
包含用于下载数据集,对数据集进行预处理和保存预测图片到文件夹的实用函数
ConvSTLM_main.py
使用· tensorflow.keras.layers.ConvLSTM2D的网络模型及其训练测试
SaConvSTLM.py
基于tensorflow实现了单元SaConvLSTM2DCell和层SaConvLSTM2D ,层SaConvLSTM2D可以直接作为模型中的一个层
SaConvSTLM_main.py
使用自定义的SaConvLSTM层SaConvLSTM的网络模型及其在MovingMNIST上的训练测试
模型结构
SaConvLSTM结构
描述
整体格局
自我注意记忆模块