DBRANet: Road Extraction by Dual-Branch Encoder and Regional Attention Decoder

被引:40
作者
Chen, Si-Bao [1 ,2 ]
Ji, Yu-Xin [1 ,2 ]
Tang, Jin [1 ,2 ]
Luo, Bin [1 ,2 ]
Wang, Wei-Qiang [3 ]
Lv, Ke [3 ]
机构
[1] Anhui Univ, Anhui Prov Key Lab Multimodal Cognit Computat, Hefei 230601, Peoples R China
[2] Anhui Univ, MOE Key Lab Intelligent Comp & Signal Proc ICSP, Hefei 230601, Peoples R China
[3] Univ Chinese Acad Sci, Sch Engn Sci, Beijing 100049, Peoples R China
基金
中国国家自然科学基金;
关键词
Attention module; convolutional neural network (CNN); remote sensing; road extraction;
D O I
10.1109/LGRS.2021.3074524
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
Although widely exploited in recent decades, road extraction is still a very significant and challenging research in the field of remote sensing image processing due to the complex background and road distribution. Among the existing CNN-based methods, U-shape architectures composed of encoders and decoders have shown their effectiveness. In this letter, we propose an improved encoder-decoder method, named DBRANet, for extracting roads from remote sensing images. In the encoding phase, we present a dual-branch network module (DBNM) to construct more effective features, thus improving the fusion feature maps of different scales. One branch utilizes the residual block, and the other branch utilizes the refined asymmetric block, which effectively increases the feature extraction capability of the backbone. In the decoding phase, considering the sinuous shape and the unbalanced distribution of roads in remote sensing images, we design a novel attention module, named the regional attention network module (RANM), to automatically learn the importance of each channel according to the regional information. Extensive experiments on several public remote sensing road data sets show that our DBRANet achieves higher segmentation [F1 score and Intersection over Union (IOU)] and connectivity [average path length similarity (APLS)] accuracy, which verifies the effectiveness of our approach.
引用
收藏
页数:5
相关论文
共 25 条
  • [11] Hu J, 2018, PROC CVPR IEEE, P7132, DOI [10.1109/TPAMI.2019.2913372, 10.1109/CVPR.2018.00745]
  • [12] Kingma D P., 2014, P INT C LEARN REPR
  • [13] Gated Auxiliary Edge Detection Task for Road Extraction With Weight-Balanced Loss
    Li, Ruirui
    Gao, Bochuan
    Xu, Qizhi
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2021, 18 (05) : 786 - 790
  • [14] Road Segmentation Based on Hybrid Convolutional Network for High-Resolution Visible Remote Sensing Image
    Li, Ye
    Guo, Lili
    Rao, Jun
    Xu, Lele
    Jin, Shan
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2019, 16 (04) : 613 - 617
  • [15] Remixed Reality: Manipulating Space and Time in Augmented Reality
    Lindlbauer, David
    Wilson, Andrew D.
    [J]. PROCEEDINGS OF THE 2018 CHI CONFERENCE ON HUMAN FACTORS IN COMPUTING SYSTEMS (CHI 2018), 2018,
  • [16] DeepRoadMapper: Extracting Road Topology from Aerial Images
    Mattyus, Gelert
    Luo, Wenjie
    Urtasun, Raquel
    [J]. 2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, : 3458 - 3466
  • [17] Mnih V., 2013, Machine Learning for Aerial Image Labeling
  • [18] U-Net: Convolutional Networks for Biomedical Image Segmentation
    Ronneberger, Olaf
    Fischer, Philipp
    Brox, Thomas
    [J]. MEDICAL IMAGE COMPUTING AND COMPUTER-ASSISTED INTERVENTION, PT III, 2015, 9351 : 234 - 241
  • [19] Rethinking the Inception Architecture for Computer Vision
    Szegedy, Christian
    Vanhoucke, Vincent
    Ioffe, Sergey
    Shlens, Jon
    Wojna, Zbigniew
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 2818 - 2826
  • [20] Non-local Neural Networks
    Wang, Xiaolong
    Girshick, Ross
    Gupta, Abhinav
    He, Kaiming
    [J]. 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 7794 - 7803