Multi-branch reverse attention semantic segmentation network for building extraction

被引:2
|
作者
Jiang, Wenxiang [1 ]
Chen, Yan [1 ]
Wang, Xiaofeng [1 ]
Kang, Menglei [1 ]
Wang, Mengyuan [1 ]
Zhang, Xuejun [2 ]
Xu, Lixiang [1 ]
Zhang, Cheng [1 ]
机构
[1] Hefei Univ, Sch Artificial Intelligence & Big Data, Hefei 230601, Peoples R China
[2] Anhui Univ, Sch Comp Sci & Technol, Hefei 230601, Peoples R China
关键词
Reverse attention; Multi-branch convolution block; Semantic segmentation; Building extraction; High spatial resolution imagery; NET;
D O I
10.1016/j.ejrs.2023.12.003
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
Extraction of color and texture features of buildings from high-resolution remote sensing images often encounters the problems of interference of background information and varying target scales. In addition, most of the current attention mechanisms focus on building key feature selection for building extraction optimization, but ignore the influence of the complex background. Hence, we propose incorporating a novel reverse attention module into the network. The innovative module enables the model to selectively extract crucial building features while suppressing the impact of intricate background noise. It mitigates the influence of uniform spectral and structurally similar heterogeneous background targets on building segmentation and extraction. As a result, the overall generalizability of the model is improved. The reverse attention can also emphasize and amplify the specific details pertaining to the boundaries of the target. Furthermore, we couple a new multi-branch convolution block into the encoder, integrating dilated convolutions with multiple dilation rates. Compared to other methods that use only one multi-scale module to extract multi-scale information from high-level features, we use different receptive field convolutions to simultaneously capture multi-scale targets from multi-level features, effectively improving the ability of the model to extract multi-scale building features. The experimental findings demonstrate that our proposed multi-branch reverse attention semantic segmentation network achieves IoU of 90.59% and 81.79% on the well-known WHU building and Inria aerial image datasets, respectively.
引用
收藏
页码:10 / 17
页数:8
相关论文
共 50 条
  • [41] Channel Attention Multi-Branch Network for Fine-Grained Image Recognition
    Wang Binzhou
    Xiao Zhiyong
    LASER & OPTOELECTRONICS PROGRESS, 2021, 58 (22)
  • [42] Automated thorax disease diagnosis using multi-branch residual attention network
    Li, Dongfang
    Huo, Hua
    Jiao, Shupei
    Sun, Xiaowei
    Chen, Shuya
    SCIENTIFIC REPORTS, 2024, 14 (01):
  • [43] Remote Sensing Scene Classification via Multi-Branch Local Attention Network
    Chen, Si-Bao
    Wei, Qing-Song
    Wang, Wen-Zhong
    Tang, Jin
    Luo, Bin
    Wang, Zu-Yuan
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2022, 31 : 99 - 109
  • [44] SaraNet: Semantic aggregation reverse attention network for pulmonary nodule segmentation
    Wang J.
    Qi M.
    Xiang Z.
    Tian Y.
    Tong D.
    Computers in Biology and Medicine, 2024, 177
  • [45] Multi-branch residual image semantic segmentation combined with inverse weight gated-control
    Qu, Haicheng
    Wang, Xiaona
    Wang, Ying
    Chen, Yao
    IMAGE AND VISION COMPUTING, 2024, 143
  • [46] Multi-Branch Regression Network For Building Classification Using Remote Sensing Images
    Gui, Yuanyuan
    Li, Xiang
    Li, Wei
    Yue, Anzhi
    2018 10TH IAPR WORKSHOP ON PATTERN RECOGNITION IN REMOTE SENSING (PRRS), 2018,
  • [47] A Multi-Branch Feature Fusion Network for Building Detection in Remote Sensing Images
    Li, Chao
    Huang, Xinyu
    Tang, Jiechen
    Wang, Kai
    IEEE ACCESS, 2021, 9 (09): : 168511 - 168519
  • [48] MBNet: Multi-Branch Network for Extraction of Rural Homesteads Based on Aerial Images
    Wei, Ren
    Fan, Beilei
    Wang, Yuting
    Zhou, Ailian
    Zhao, Zijuan
    REMOTE SENSING, 2022, 14 (10)
  • [49] Attention-modulated multi-branch convolutional neural networks for neonatal brain tissue segmentation
    Fan, Xunli
    Shan, Shixi
    Li, Xianjun
    Li, Jinhang
    Mi, Jizong
    Yang, Jian
    Zhang, Yongqin
    COMPUTERS IN BIOLOGY AND MEDICINE, 2022, 146
  • [50] Two-branch encoding and iterative attention decoding network for semantic segmentation
    Zhu, Hegui
    Zhang, Min
    Zhang, Xiangde
    Zhang, Libo
    NEURAL COMPUTING & APPLICATIONS, 2021, 33 (10): : 5151 - 5166