摘要
针对高分辨率遥感图像语义分割面临的边界模糊、错分和漏分割问题,提出一种改进的pix2pix模型用于遥感图像语义分割。首先,在分割网络编码中,通过引入空洞空间金字塔池化模块来提取不同尺度的图像上下文特征;然后,在跳跃连接阶段引入空间注意力机制对地物边缘细节信息进行增强,以提高网络模型对地物的分割能力。将所提出的方法在Vaihingen和Gaofen Image Dataset(GID)数据集上进行语义分割仿真实验,mIoU分别达到82.69%和81.27%。仿真实验结果表明:所提出的方法充分地利用了上下文信息,有效减少了错误分割,使分割边界更清晰,优于其它经典语义分割方法。
- 单位