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 条
  • [31] SAR ship detection based on salience region extraction and multi-branch attention
    Zha, Cheng
    Min, Weidong
    Han, Qing
    Xiong, Xin
    Wang, Qi
    Xiang, Hongyue
    INTERNATIONAL JOURNAL OF APPLIED EARTH OBSERVATION AND GEOINFORMATION, 2023, 123
  • [32] Multi-Scale and Multi-Branch Convolutional Neural Network for Retinal Image Segmentation
    Jiang, Yun
    Liu, Wenhuan
    Wu, Chao
    Yao, Huixiao
    SYMMETRY-BASEL, 2021, 13 (03): : 1 - 25
  • [33] Semantic-aware multi-branch interaction network for deep multimodal learning
    Hao Pan
    Jun Huang
    Neural Computing and Applications, 2023, 35 : 7529 - 7545
  • [34] Object Proposal by Multi-branch Hierarchical Segmentation
    Wang, Chaoyang
    Zhao, Long
    Liang, Shuang
    Zhang, Liqing
    Jia, Jinyuan
    Wei, Yichen
    2015 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2015, : 3873 - 3881
  • [35] Semantic-aware multi-branch interaction network for deep multimodal learning
    Pan, Hao
    Huang, Jun
    NEURAL COMPUTING & APPLICATIONS, 2023, 35 (10): : 7529 - 7545
  • [36] Deep Convolutional Neural Network for Segmentation and Classification of Structural Multi-branch Cracks
    Kandula, Himavanth
    Koduri, Hrushith Ram
    Kalapatapu, Prafulla
    Pasupuleti, Venkata Dilip Kumar
    EUROPEAN WORKSHOP ON STRUCTURAL HEALTH MONITORING (EWSHM 2022), VOL 2, 2023, : 177 - 185
  • [37] Facial Expression Recognition Using Multi-Branch Attention Convolutional Neural Network
    He, Yinggang
    IEEE ACCESS, 2023, 11 : 1244 - 1253
  • [38] Image super-resolution network based on a multi-branch attention mechanism
    Xin Yang
    Yingqing Guo
    Zhiqiang Li
    Dake Zhou
    Signal, Image and Video Processing, 2021, 15 : 1397 - 1405
  • [39] MIINet: a multi-branch information interaction network for few-shot segmentation
    Zhang, Zhaopeng
    Xu, Zhijie
    Zhang, Jianqin
    SIGNAL IMAGE AND VIDEO PROCESSING, 2024, 18 (12) : 9081 - 9094
  • [40] Image super-resolution network based on a multi-branch attention mechanism
    Yang, Xin
    Guo, Yingqing
    Li, Zhiqiang
    Zhou, Dake
    SIGNAL IMAGE AND VIDEO PROCESSING, 2021, 15 (07) : 1397 - 1405