概述 自我时空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结构 描述 整体格局 自我注意记忆模块