一种交叉区域注意力的高分辨率遥感建筑物提取算法

作者：邓博文; 徐胜军^*; 孟月波; 刘光辉; 韩九强; 史亚

来源：小型微型计算机系统, 2023, 1-8.

• 遥感图像
• 建筑物分割
• 交叉区域注意力
• 通道注意力

针对遥感图像中建筑物区域尺度跨度大且区域边界模糊导致分割精度低的问题，本文提出了一种基于交叉区域注意力的遥感建筑物分割算法。首先，设计了交叉自注意力模块和分组通道注意力模块用于建立遥感图像区域间和区域内特征的相关性表征，进而引导模型关注待分割目标的区域级细节特征与通道组选择能力；最后，针对分割结果缺乏空间相关性约束问题，提出一种区域一致性监督的损失函数，约束局部区域内像素标签分配的一致性。所提算法在WHU数据集上IoU、Precision、Recall、F1-score分别可达到91.2%、 95.28%、95.4%和95.3%；在Massachusetts数据集上IoU、Precision、Recall、F1-score分别可达到74.6%、83.7%、86.9%和85.3%，各项指标均优于主流遥感图像建筑物分割算法。

• 单位
  西安交通大学; 西安建筑科技大学