• MU-UNet:融合层次化编码与多尺度输出的遥感建筑物分割

    MU-UNet: remote sensing building segmentation by integrating hierarchical encoding and multi-scale output

    • 针对遥感影像建筑物分割任务中存在的边界细节分割不完整及小型建筑物易漏检等问题,本文提出一种改进的U-Net建筑物提取网络——MU-UNet。该方法以经典U-Net框架为基础,引入Hiera编码器与基于密集连接的多尺度输出结构,以增强跨层级上下文信息建模能力;结合SE模块提升通道间特征的自适应表达能力;同时设计卷积边缘焦点模块(Convolutional Edge-Focal Module,CEFM),强化细粒度边界建模与特征融合能力。实验结果表明,该方法在WHU航空影像数据集、WHU卫星数据集及Massachusetts数据集上的交并比(Intersection over Union, IoU)和F1指标均优于主流模型。其中,在WHU航空影像数据集上,MU-UNet相较于基线U-Net的IoU与F1分别提升3.65%和2.04%;在Massachusetts数据集上分别提升4.99%和3.47%。

       

      Abstract: Addressing the issues of incomplete boundary segmentation and frequent omission of small-scale buildings in remote sensing image building extraction tasks, this paper proposes an enhanced U-Net-based building extraction network named MU-UNet. Building upon the classical U-Net architecture, the proposed method incorporates a Hiera encoder and a densely connected multi-scale output structure to strengthen cross-hierarchical contextual modeling capabilities. It further integrates Squeeze-and-Excitation (SE) modules to enhance adaptive feature representation across channels. Additionally, a Convolutional Edge-Focal Module (CEFM) module is designed to reinforce fine-grained boundary modeling and cross-scale feature fusion. The experimental results show that the method proposed in this paper outperforms the mainstream models in terms of both IoU and F1 metrics on the WHU aerial image dataset, the WHU satellite dataset, and the Massachusetts dataset. Specifically, on the WHU aerial image dataset, MU-UNet improves IoU(Intersection over Union) and F1 by 3.65% and 2.04% respectively compared to the baseline U-Net; on the Massachusetts dataset, the improvements are 4.99% and 3.47% respectively.

       

    /

    返回文章
    返回