基于Transformer-IRB的短临降雨预测方法、装置和存储介质

摘要

本发明公开了一种基于Transformer‑IRB的短临降雨预测方法、装置和存储介质，方法包括：将雷达回波图像进行预处理，将预处理之后的连续的M帧雷达回波图像送入预先构建并训练好的基于Transformer‑IRB的短临降雨预测模型以得到N帧雷达回波图像，短临降雨预测模型包括块嵌入模块、特征提取模块、解码模块，块嵌入模块提取连续的多帧雷达回波图的嵌入特征并加上位置编码后送入特征提取模块，特征提取模块采用若干具有多头注意力层的Transformer‑倒置残差块模块构成编码器对输入的嵌入特征进行特征提取后送入解码模块，解码模块对输入进行解码得到连续多帧雷达回波图，本发明得以更好拟合雷达回波图序列中长范围时空范围的降雨演变过程，提升预测精度。