IDANet: Iterative D-LinkNets with Attention for Road Extraction from High-Resolution Satellite Imagery

被引:1
作者
Xu, Benzhu [1 ]
Bao, Shengshuai [1 ]
Zheng, Liping [1 ]
Zhang, Gaofeng [1 ]
Wu, Wenming [1 ]
机构
[1] Hefei Univ Technol, Sch Comp Sci & Informat Engn, Hefei 230009, Peoples R China
来源
PATTERN RECOGNITION AND COMPUTER VISION, PRCV 2021, PT II | 2021年 / 13020卷
基金
中国国家自然科学基金;
关键词
Road extraction; Semantic segmentation; Convolutional neural network; Attention mechanism; Iterative architecture;
D O I
10.1007/978-3-030-88007-1_12
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Road information plays a fundamental role in many application fields, while satellite images are able to capture a large area of the ground with high resolution. Therefore, extracting roads has become a hot research topic in the field of remote sensing. In this paper, we propose a novel semantic segmentation model, named IDANet, which adopts iterative D-LinkNets with attention modules for road extraction from high-resolution satellite images. Our road extraction model is built on D-LinkNet, an effective network which adopts encoder-decoder structure, dilated convolution, and pretrained encoder for road extraction task. The attention mechanism can be used to achieve a better fusion of features from different levels. To this end, a modified D-LinkNet with attention is proposed for more effective feature extraction. With this network as the basic refinement module, we further adopt an iterative architecture to maximize the network performance, where the output of the previous network serves as the input of the next network to refine the road segmentation and obtain enhanced results. The evaluation demonstrates the superior performance of our proposed model. Specifically, the performance of our model exceeds the original D-LinkNet by 2.2% of the IoU on the testing dataset of DeepGlobe for road extraction.
引用
收藏
页码:140 / 152
页数:13
相关论文
共 22 条
  • [1] Deep Learning Approaches Applied to Remote Sensing Datasets for Road Extraction: A State-Of-The-Art Review
    Abdollahi, Abolfazl
    Pradhan, Biswajeet
    Shukla, Nagesh
    Chakraborty, Subrata
    Alamri, Abdullah
    [J]. REMOTE SENSING, 2020, 12 (09)
  • [2] SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation
    Badrinarayanan, Vijay
    Kendall, Alex
    Cipolla, Roberto
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (12) : 2481 - 2495
  • [3] RoadTracer: Automatic Extraction of Road Networks from Aerial Images
    Bastani, Favyen
    He, Songtao
    Abbar, Sofiane
    Alizadeh, Mohammad
    Balakrishnan, Hari
    Chawla, Sanjay
    Madden, Sam
    DeWitt, David
    [J]. 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 4720 - 4728
  • [4] Chaurasia A, 2017, 2017 IEEE VISUAL COMMUNICATIONS AND IMAGE PROCESSING (VCIP)
  • [5] Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation
    Chen, Liang-Chieh
    Zhu, Yukun
    Papandreou, George
    Schroff, Florian
    Adam, Hartwig
    [J]. COMPUTER VISION - ECCV 2018, PT VII, 2018, 11211 : 833 - 851
  • [6] DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs
    Chen, Liang-Chieh
    Papandreou, George
    Kokkinos, Iasonas
    Murphy, Kevin
    Yuille, Alan L.
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2018, 40 (04) : 834 - 848
  • [7] DeepGlobe 2018: A Challenge to Parse the Earth through Satellite Images
    Demir, Ilke
    Koperski, Krzysztof
    Lindenbaum, David
    Pang, Guan
    Huang, Jing
    Bast, Saikat
    Hughes, Forest
    Tuia, Devis
    Raskar, Ramesh
    [J]. PROCEEDINGS 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW), 2018, : 172 - 181
  • [8] A Brief Survey on Semantic Segmentation with Deep Learning
    Hao, Shijie
    Zhou, Yuan
    Guo, Yanrong
    [J]. NEUROCOMPUTING, 2020, 406 : 302 - 321
  • [9] Hu J, 2018, PROC CVPR IEEE, P7132, DOI [10.1109/TPAMI.2019.2913372, 10.1109/CVPR.2018.00745]
  • [10] Kang Huang, 2020, Pattern Recognition and Computer Vision. Third Chinese Conference, PRCV 2020. Proceedings. Lecture Notes in Computer Science (LNCS 12305), P315, DOI 10.1007/978-3-030-60633-6_26