一种雷达回波气象数据仿真预测生成方法、装置、电子设备及存储介质

本发明公开了一种雷达回波气象数据仿真预测生成方法、装置、电子设备及存储介质，首先收集真实的雷达回波视频及视频的文本标注作为训练集；构建RoBERTa模型对文本标注数据进行特征提取；针对收集的雷达回波视频数据，利用背景差分法和帧间差分法提取帧间变化信息；构建基于Transformer的图像编码器对初始帧提取空间特征；搭建扩散模型，将初始帧特征及文本特征融合，作为条件信息输入扩散模型，将帧间变化序列输入扩散模型进行前向扩散；对扩散模型进行反向去噪，预测所加噪声计算损失，训练扩散模型；将测试样本输入训练好的扩散模型去噪采样获得预测的帧间变化序列，将预测的帧间变化序列与测试样本初始帧结合，预测生成未来的雷达回波视频。