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 条
  • [1] EfficientViT: Lightweight Multi-Scale Attention for High-Resolution Dense Prediction
    Cai, Han
    Li, Junyan
    Hu, Muyan
    Gan, Chuang
    Han, Song
    2023 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2023), 2023, : 17256 - 17267
  • [2] Multi-Scale Feature Extraction Method of Hyperspectral Image with Attention Mechanism
    Xu Zhangchi
    Guo Baofeng
    Wu Wenhao
    You Jingyun
    Su Xiaotong
    LASER & OPTOELECTRONICS PROGRESS, 2024, 61 (04)
  • [3] A Road Extraction Method of a High-Resolution Remote Sensing Image Based on Multi-Feature Fusion and the Attention Mechanism
    Jiang, Na
    Li, Jiyuan
    Yang, Jingyu
    Lin, Junting
    Lu, Baopeng
    TRAITEMENT DU SIGNAL, 2022, 39 (06) : 1907 - 1916
  • [4] A lightweight Large-Scale RS image village extraction method combining deep transitive transfer learning and attention mechanism
    Liu, Yang
    Zhao, Quanhua
    Jia, Shuhan
    Li, Yu
    EGYPTIAN JOURNAL OF REMOTE SENSING AND SPACE SCIENCES, 2024, 27 (02): : 192 - 203
  • [5] HRU-Net: High-Resolution Remote Sensing Image Road Extraction Based on Multi-Scale Fusion
    Yin, Anchao
    Ren, Chao
    Yan, Zhiheng
    Xue, Xiaoqin
    Yue, Weiting
    Wei, Zhenkui
    Liang, Jieyu
    Zhang, Xudong
    Lin, Xiaoqi
    APPLIED SCIENCES-BASEL, 2023, 13 (14):
  • [6] Multi-scale attention integrated hierarchical networks for high-resolution building footprint extraction
    Liu, Tang
    Yao, Ling
    Qin, Jun
    Lu, Ning
    Jiang, Hou
    Zhang, Fan
    Zhou, Chenghu
    INTERNATIONAL JOURNAL OF APPLIED EARTH OBSERVATION AND GEOINFORMATION, 2022, 109
  • [7] A lightweight multi-scale channel attention network for image super-resolution
    Li, Wenbin
    Li, Juefei
    Li, Jinxin
    Huang, Zhiyong
    Zhou, Dengwen
    NEUROCOMPUTING, 2021, 456 : 327 - 337
  • [8] Lightweight Multi-Scale Asymmetric Attention Network for Image Super-Resolution
    Zhang, Min
    Wang, Huibin
    Zhang, Zhen
    Chen, Zhe
    Shen, Jie
    MICROMACHINES, 2022, 13 (01)
  • [9] Multi-scale convolutional attention network for lightweight image super-resolution
    Xie, Feng
    Lu, Pei
    Liu, Xiaoyong
    JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2023, 95
  • [10] Lightweight multi-scale residual networks with attention for image super-resolution
    Liu, Huan
    Cao, Feilong
    Wen, Chenglin
    Zhang, Qinghua
    KNOWLEDGE-BASED SYSTEMS, 2020, 203