Remote Sensing Technology and Application, Volume. 39, Issue 3, 620(2024)

Coupling Mask R-CNN and Attention Mechanism for Building Extraction and Post-Processing Strategy

Buyu SU, Xiaoping DU, Haowei MU, Chen XU, Fang CHEN, and Xiaonan LUO
Author Affiliations
  • School of Computer Science and Information SecurityGuilin University of Electronic Technology, Guilin541004,China
  • show less

    Buildings are integral components of urban areas. Extracting buildings from high-resolution remote sensing data holds significant academic importance in areas such as land use analysis, urban planning, and disaster risk reduction. For the problems of building extraction, an improved Mask R-CNN building instance segmentation model is proposed. Based on the residual neural network fusion convolutional attention model, a residual convolutional attention network is constructed to improve the problem of inadequate feature extraction. The loss function is optimized by adding the Dice Loss method, and then the feature learning process is optimized. And a post-processing strategy combining Douglas-Peucker algorithm and Fine polygon regularization algorithm is introduced to make the building contours more regular and smooth. The experimental results show that the improved model improves the detection mAP value by 7.74% at Iou 0.5 and 7.57% at Iou 0.75 compared with the original model, and the post-processing strategy improves the F1-Score value by 6.01% compared with the original model after selecting the appropriate threshold to optimize the mask. The instance segmentation model coupled with Mask R-CNN and attention mechanism improves the small building misdetection and omission problem, building segmentation boundary adhesion problem, and building segmentation accuracy; building post-processing strategy, improves building regularization.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Buyu SU, Xiaoping DU, Haowei MU, Chen XU, Fang CHEN, Xiaonan LUO. Coupling Mask R-CNN and Attention Mechanism for Building Extraction and Post-Processing Strategy[J]. Remote Sensing Technology and Application, 2024, 39(3): 620

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Nov. 9, 2022

    Accepted: --

    Published Online: Dec. 9, 2024

    The Author Email:

    DOI:10.11873/j.issn.1004-0323.2024.3.0620

    Topics