扩散模型训练方法、遥感图像生成方法、电子设备与计算机可读存储介质

本申请提供一种扩散模型的训练方法，包括：获取遥感样本集，基于样本图像生成对应的内容文本；利用SpDINO模型作为编码器将样本图像编码到潜空间，得到样本特征，其中，SpDINO模型是使用DINOv2、SAM2和CLIP对ViT进行多教师蒸馏训练得到；对样本特征进行加噪处理得到含噪特征；利用基于Transformer架构的扩散模型基于内容文本和去噪时间步对含噪特征进行扩散预测，得到预测特征，针对各个去噪时间步的含噪特征得到对应的预测特征；基于含噪特征和预测特征计算扩散模型损失值，并根据扩散模型损失值调节扩散模型的参数，得到训练好的扩散模型。本申请通过使用多教师蒸馏训练得到的SpDINO模型作为编码器对扩散模型进行训练，使得扩散模型能够生成特征完善、清晰度高的图像。