A Lightweight High-Resolution RS Image Road Extraction Method Combining Multi-Scale and Attention Mechanism

被引:4
|
作者
Wang, Rui [1 ]
Cai, Mingxiang [1 ]
Xia, Zixuan [2 ]
机构
[1] China Transport Telecommun & Informat Ctr, Beijing 100011, Peoples R China
[2] Heilongjiang Univ Technol, Coll Art & Architectural Engn, Jixi 158100, Peoples R China
关键词
Road extraction; deep learning; CAM; SAM; ASPP; lightweight; SEGMENTATION;
D O I
10.1109/ACCESS.2023.3313390
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Road information plays an indispensable role in human society's development. However, owing to the diversity and complexity of roads, it is difficult to obtain satisfactory road-extraction result. Some typical factors, such as discontinuity, loss of edge details, and long-time consumption, have negative impacts on obtaining accurate road information. These problems are particularly prominent during road extraction when high-resolution remote-sensing images are used. To obtain accurate road information, a novel lightweight deep learning neural network was pro-posed in this study by integrating a multiscale module and attention mechanisms. As an excellent multiscale segmentation module, the atrous spatial pyramid pooling was selected to enhance the road extraction ability of remote sensing images. In addition, an attention mechanism was employed to solve the problems of discontinuity and loss of edge details in road extraction, and MobileNet V2 was selected as the backbone of DeepLab V3+ because of its lightweight structure, which can help solve the problem of excessive training time consumption. The experimental verification was carried out on the Ottawa road dataset and the Massachusetts road dataset. Experimental results show that compared with U-Net, SegNet and MDeeplab v3+ networks, the proposed algorithm is the best in IoU, Recall, OA and Kappa. Among them, on the Ottawa road dataset, the OA and Kappa of the algorithm in this paper are 98.92 % and 95.02 %, respectively. On the Massachusetts road dataset, OA and Kappa 98.29% and 89.87%. In addition, the training time was significantly shorter than that of the other deep learning networks. The proposed method exhibited a good performance in road extraction.
引用
收藏
页码:108956 / 108966
页数:11
相关论文
共 50 条
  • [31] ROAD DAMAGE DETECTION FROM HIGH-RESOLUTION RS IMAGE
    Gong, Lixia
    An, Liqiang
    Liu, Mingzhong
    Zhang, Jingfa
    2012 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS), 2012, : 990 - 993
  • [32] Multi-scale strip-shaped convolution attention network for lightweight image super-resolution
    Xu, Ke
    Pan, Lulu
    Peng, Guohua
    Zhang, Wenbo
    Lv, Yanheng
    Li, Guo
    Li, Lingxiao
    Lei, Le
    SIGNAL PROCESSING-IMAGE COMMUNICATION, 2024, 128
  • [33] Multi-scale attention network for image super-resolution
    Wang, Li
    Shen, Jie
    Tang, E.
    Zheng, Shengnan
    Xu, Lizhong
    JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2021, 80
  • [34] Multi-Scale Fusion Siamese Network Based on Three-Branch Attention Mechanism for High-Resolution Remote Sensing Image Change Detection
    Li, Yan
    Weng, Liguo
    Xia, Min
    Hu, Kai
    Lin, Haifeng
    REMOTE SENSING, 2024, 16 (10)
  • [35] High-Resolution Image Inpainting using Multi-Scale Neural Patch Synthesis
    Yang, Chao
    Lu, Xin
    Lin, Zhe
    Shechtman, Eli
    Wang, Oliver
    Li, Hao
    30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 4076 - 4084
  • [36] DSMSA-Net: Deep Spatial and Multi-scale Attention Network for Road Extraction in High Spatial Resolution Satellite Images
    Khan, Sultan Daud
    Alarabi, Louai
    Basalamah, Saleh
    ARABIAN JOURNAL FOR SCIENCE AND ENGINEERING, 2023, 48 (02) : 1907 - 1920
  • [37] DSMSA-Net: Deep Spatial and Multi-scale Attention Network for Road Extraction in High Spatial Resolution Satellite Images
    Sultan Daud Khan
    Louai Alarabi
    Saleh Basalamah
    Arabian Journal for Science and Engineering, 2023, 48 : 1907 - 1920
  • [38] PCB defects target detection combining multi-scale and attention mechanism
    Jiang, Wujin
    Li, Taifu
    Zhang, Shaolin
    Chen, Wenbin
    Yang, Jie
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2023, 123
  • [39] Multi-scale salient object detection network combining an attention mechanism
    Liu, Di
    Guo, Jichang
    Wang, Yudong
    Zhang, Yi
    Xi'an Dianzi Keji Daxue Xuebao/Journal of Xidian University, 2022, 49 (04): : 118 - 126
  • [40] Rice Ears Detection Method Based on Multi-Scale Image Recognition and Attention Mechanism
    Qiu, Fen
    Shen, Xiaojun
    Zhou, Cheng
    He, Wuming
    Yao, Lili
    IEEE ACCESS, 2024, 12 : 68637 - 68647