Road Extraction by Using Atrous Spatial Pyramid Pooling Integrated Encoder-Decoder Network and Structural Similarity Loss

被引:136
作者
He, Hao [1 ]
Yang, Dongfang [1 ]
Wang, Shicheng [1 ]
Wang, Shuyang [2 ]
Li, Yongfei [1 ]
机构
[1] Rocket Force Univ Engn, Dept Control Engn, Xian 710025, Shaanxi, Peoples R China
[2] Rocket Force Univ Engn, Dept Informat Engn, Xian 710025, Shaanxi, Peoples R China
基金
中国国家自然科学基金;
关键词
deep learning; remote sensing; road extraction; semantic segmentation; structural similarity;
D O I
10.3390/rs11091015
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
The technology used for road extraction from remote sensing images plays an important role in urban planning, traffic management, navigation, and other geographic applications. Although deep learning methods have greatly enhanced the development of road extractions in recent years, this technology is still in its infancy. Because the characteristics of road targets are complex, the accuracy of road extractions is still limited. In addition, the ambiguous prediction of semantic segmentation methods also makes the road extraction result blurry. In this study, we improved the performance of the road extraction network by integrating atrous spatial pyramid pooling (ASPP) with an Encoder-Decoder network. The proposed approach takes advantage of ASPP's ability to extract multiscale features and the Encoder-Decoder network's ability to extract detailed features. Therefore, it can achieve accurate and detailed road extraction results. For the first time, we utilized the structural similarity (SSIM) as a loss function for road extraction. Therefore, the ambiguous predictions in the extraction results can be removed, and the image quality of the extracted roads can be improved. The experimental results using the Massachusetts Road dataset show that our method achieves an F1-score of 83.5% and an SSIM of 0.893. Compared with the normal U-net, our method improves the F1-score by 2.6% and the SSIM by 0.18. Therefore, it is demonstrated that the proposed approach can extract roads from remote sensing images more effectively and clearly than the other compared methods.
引用
收藏
页数:16
相关论文
共 35 条
  • [1] Simultaneous extraction of roads and buildings in remote sensing imagery with convolutional neural networks
    Alshehhi, Rasha
    Marpu, Prashanth Reddy
    Woon, Wei Lee
    Dalla Mura, Mauro
    [J]. ISPRS JOURNAL OF PHOTOGRAMMETRY AND REMOTE SENSING, 2017, 130 : 139 - 149
  • [2] 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
  • [3] Chen LB, 2017, IEEE INT SYMP NANO, P1, DOI 10.1109/NANOARCH.2017.8053709
  • [4] Clevert D.-A., 2016, 4 INT C LEARN REPR I
  • [5] Goodfellow I, 2016, ADAPT COMPUT MACH LE, P1
  • [6] Automatic Road Detection and Centerline Extraction via Cascaded End-to-End Convolutional Neural Network
    Cheng, Guangliang
    Wang, Ying
    Xu, Shibiao
    Wang, Hongzhen
    Xiang, Shiming
    Pan, Chunhong
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2017, 55 (06): : 3322 - 3337
  • [7] [贺浩 He Hao], 2019, [测绘学报, Acta Geodetica et Cartographica Sinica], V48, P330
  • [8] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778
  • [9] Road network extraction and intersection detection from aerial images by tracking road footprints
    Hu, Jiuxiang
    Razdan, Anshuman
    Femiani, John C.
    Cui, Ming
    Wonka, Peter
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2007, 45 (12): : 4144 - 4157
  • [10] Iglovikov V., 2017, ARXIVLABS COMPUTER V