Seg-Road: A Segmentation Network for Road Extraction Based on Transformer and CNN with Connectivity Structures

被引:33
|
作者
Tao, Jingjing [1 ]
Chen, Zhe [2 ,3 ,4 ]
Sun, Zhongchang [3 ,4 ,5 ]
Guo, Huadong [3 ,4 ,5 ]
Leng, Bo [6 ]
Yu, Zhengbo [2 ]
Wang, Yanli [2 ]
He, Ziqiong [6 ]
Lei, Xiangqi [7 ]
Yang, Jinpei [8 ]
机构
[1] Tianjin Normal Univ, Coll Geog & Environm Sci, Tianjin 300387, Peoples R China
[2] Chengdu Univ Technol, Coll Math & Phys, Chengdu 610059, Peoples R China
[3] Int Res Ctr Big Data Sustainable Dev Goals CBAS, Beijing 100094, Peoples R China
[4] Chinese Acad Sci, Aerosp Informat Res Inst, Key Lab Digital Earth Sci, Beijing 100094, Peoples R China
[5] Chinese Acad Sci, Aerosp Informat Res Inst, Hainan Lab Earth Observat, Sanya 572029, Peoples R China
[6] Chengdu Univ Technol, Coll Management Sci, Chengdu 610059, Peoples R China
[7] Chengdu Univ Technol, Coll Earth Sci, Chengdu 610059, Peoples R China
[8] Nanjing Univ, Coll Architecture & Urban Planning, Nanjing 210093, Peoples R China
基金
中国国家自然科学基金;
关键词
road extraction; remote sensing imagery; pixel connectivity structure; Seg-Road;
D O I
10.3390/rs15061602
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
Acquiring road information is important for smart cities and sustainable urban development. In recent years, significant progress has been made in the extraction of urban road information from remote sensing images using deep learning (DL) algorithms. However, due to the complex shape, narrowness, and high span of roads in the images, the results are often unsatisfactory. This article proposes a Seg-Road model to improve road connectivity. The Seg-Road uses a transformer structure to extract the long-range dependency and global contextual information to improve the fragmentation of road segmentation and uses a convolutional neural network (CNN) structure to extract local contextual information to improve the segmentation of road details. Furthermore, a novel pixel connectivity structure (PCS) is proposed to improve the connectivity of road segmentation and the robustness of prediction results. To verify the effectiveness of Seg-Road for road segmentation, the DeepGlobe and Massachusetts datasets were used for training and testing. The experimental results show that Seg-Road achieves state-of-the-art (SOTA) performance, with an intersection over union (IoU) of 67.20%, mean intersection over union (MIoU) of 82.06%, F1 of 91.43%, precision of 90.05%, and recall of 92.85% in the DeepGlobe dataset, and achieves an IoU of 68.38%, MIoU of 83.89%, F1 of 90.01%, precision of 87.34%, and recall of 92.86% in the Massachusetts dataset, which is better than the values for CoANet. Further, it has higher application value for achieving sustainable urban development.
引用
收藏
页数:21
相关论文
共 50 条
  • [21] Road Segmentation Using CNN and Distributed LSTM
    Lyu, Yecheng
    Bai, Lin
    Huang, Xinming
    2019 IEEE INTERNATIONAL SYMPOSIUM ON CIRCUITS AND SYSTEMS (ISCAS), 2019,
  • [22] DSNet: an efficient CNN for road scene segmentation
    Chen, Ping-Rong
    Hang, Hsueh-Ming
    Chan, Sheng-Wei
    Lin, Jing-Jhih
    APSIPA TRANSACTIONS ON SIGNAL AND INFORMATION PROCESSING, 2020, 9 (01) : 1 - 14
  • [23] CoANet: Connectivity Attention Network for Road Extraction From Satellite Imagery
    Mei, Jie
    Li, Rou-Jing
    Gao, Wang
    Cheng, Ming-Ming
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2021, 30 (30) : 8540 - 8552
  • [24] SwinURNet: Hybrid Transformer-CNN Architecture for Real-Time Unstructured Road Segmentation
    Wang, Zhangyu
    Liao, Zhihao
    Zhou, Bin
    Yu, Guizhen
    Luo, Wenwen
    IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2024, 73
  • [25] U-Net-Based CNN Architecture for Road Crack Segmentation
    Di Benedetto, Alessandro
    Fiani, Margherita
    Gujski, Lucas Matias
    INFRASTRUCTURES, 2023, 8 (05)
  • [26] Enhancement of the Super Pixel-CNN Based Road Segmentation Using Cycle Consistent Adversarial Network
    zohourian, Farnoush
    Pauli, Josef
    FOURTEENTH INTERNATIONAL CONFERENCE ON MACHINE VISION (ICMV 2021), 2022, 12084
  • [27] Road Traffic Sign Detection Method Based on RTS R-CNN Instance Segmentation Network
    Zhang, Guirong
    Peng, Yiming
    Wang, Hai
    SENSORS, 2023, 23 (14)
  • [28] A Context-Aware Road Extraction Method for Remote Sensing Imagery Based on Transformer Network
    Zhang, Xiaokai
    Ma, Xianzhi
    Yang, Zhigang
    Liu, Xilin
    Chen, Zehua
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2023, 20
  • [29] Image feature based GPS trace filtering for road network generation and road segmentation
    Yuan, Jiangye
    Cheriyadat, Anil M.
    MACHINE VISION AND APPLICATIONS, 2016, 27 (01) : 1 - 12
  • [30] Image feature based GPS trace filtering for road network generation and road segmentation
    Jiangye Yuan
    Anil M. Cheriyadat
    Machine Vision and Applications, 2016, 27 : 1 - 12