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 条
  • [21] Lightweight Attention Network for Very High-Resolution Image Semantic Segmentation
    Guan, Renchu
    Wang, Mingming
    Bruzzone, Lorenzo
    Zhao, Haishi
    Yang, Chen
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [22] TBNet: Stereo Image Super-Resolution with Multi-Scale Attention
    Zhu, Jiyang
    Han, Xue
    JOURNAL OF CIRCUITS SYSTEMS AND COMPUTERS, 2023, 32 (18)
  • [23] Multi-featured multi-scale combination of high-resolution remote sensing images for building extraction
    Niu, Yuhan
    APPLIED MATHEMATICS AND NONLINEAR SCIENCES, 2023, 9 (01)
  • [24] Enhanced Single Image Super Resolution Method Using Lightweight Multi-Scale Channel Dense Network
    Lee, Yooho
    Jun, Dongsan
    Kim, Byung-Gyu
    Lee, Hunjoo
    SENSORS, 2021, 21 (10)
  • [25] LMSN:a lightweight multi-scale network for single image super-resolution
    Yiye Zou
    Xiaomin Yang
    Marcelo Keese Albertini
    Farhan Hussain
    Multimedia Systems, 2021, 27 : 845 - 856
  • [26] LMSN:a lightweight multi-scale network for single image super-resolution
    Zou, Yiye
    Yang, Xiaomin
    Albertini, Marcelo Keese
    Hussain, Farhan
    MULTIMEDIA SYSTEMS, 2021, 27 (04) : 845 - 856
  • [27] Multi-scale feature selection network for lightweight image super-resolution
    Li, Minghong
    Zhao, Yuqian
    Zhang, Fan
    Luo, Biao
    Yang, Chunhua
    Gui, Weihua
    Chang, Kan
    NEURAL NETWORKS, 2024, 169 : 352 - 364
  • [28] Multi-scale hand segmentation method based on attention mechanism
    Zhou, Wenqing
    Dai, Sumin
    Wang, Yangpin
    Wang, Wenrun
    CHINESE JOURNAL OF LIQUID CRYSTALS AND DISPLAYS, 2024, 39 (11) : 1506 - 1518
  • [29] AUTOMATIC EXTRACTION OF WATER IN HIGH-RESOLUTION SAR IMAGES BASED ON MULTI-SCALE LEVEL SET METHOD AND OTSU ALGORITHM
    Sui, H. G.
    Xu, C.
    XXII ISPRS CONGRESS, TECHNICAL COMMISSION VII, 2012, 39 (B7): : 453 - 457
  • [30] Multi-scale Underwater Image Enhancement Network Based on Attention Mechanism
    Fang Ming
    Liu Xiaohan
    Fu Feiran
    JOURNAL OF ELECTRONICS & INFORMATION TECHNOLOGY, 2021, 43 (12) : 3513 - 3521