红外降水预测及模型训练方法、装置、电子设备

本公开涉及深度学习技术领域，具体涉及公开了一种红外降水预测及模型训练方法、装置、电子设备，该方法包括：获取训练集，训练集中每个样本包括一个样本区域的多通道亮温特征及其对应的降水特征真值；获取预设的红外降水预测模型，红外降水预测模型包括输入层、编码器、中间层、解码器和输出层，其中，编码器在每次下采样前增加注意力机制模块，中间层包括空洞空间金字塔池化模块，解码器在每次上采样后增加注意力机制模块，模型的加权均方误差损失函数根据降水空间异质性动态调整误差权重；使用训练集对预设的红外降水预测模型进行训练，得到训练好的红外降水预测模型。该技术方案可以更准确地进行红外降水估计。