一种基于扩散模型的高分辨率遥感图像语义分割方法

本发明属于空间光学遥感技术领域，涉及一种基于扩散模型的高分辨率遥感图像语义分割方法。具体过程为：第一阶段训练：对高分辨率遥感图像每一像素所属类型进行标注生成标签图像，利用标签图像对包含编码器和解码器的自编码器进行训练；第二阶段训练：在自编码器上基础上，加载并训练条件扩散模型，包括噪声注入模块、条件编码模块以及去噪U‑Net，在训练时将自编码器参数固定；其中，噪声注入模块对所述编码器输出的隐变量加噪，条件编码模块用于对所述高分辨率图像进行多尺度特征提取，去噪U‑Net在所述多尺度特征引导下，进行图像分量与噪声分量预测以及图像重建；所述解码器用于对重建的图像进行解码；图像语义分割：利用训练好的网络对高分辨率遥感图像进行语义分割。