A method for landslide identification and detection in high-precision aerial imagery: progressive CBAM-U-net model

被引:0
作者
Lin, Hanjie [1 ]
Li, Li [1 ]
Qiang, Yue [1 ]
Xu, Xinlong [1 ]
Liang, Siyu [1 ]
Chen, Tao [1 ]
Yang, Wenjun [1 ]
Zhang, Yi [1 ]
机构
[1] Chongqing Three Gorges Univ, Dept Civil Engn, Wanzhou 404100, Chongqing, Peoples R China
关键词
Deep learning; U-net; Attention mechanism; Landslide boundary recognition and detection; High-precision aerial landslide imagery; ALGORITHM;
D O I
10.1007/s12145-024-01465-6
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
Rapid identification and detection of landslides is of significance for disaster damage assessment and post-disaster relief. However, U-net for rapid landslide identification and detection suffers from semantic gap and loss of spatial information. For this purpose, this paper proposed the U-net with a progressive Convolutional Block Attention Module (CBAM-U-net) for landslide boundary identification and extraction from high-precision aerial imagery. Firstly, 109 high-precision aerial landslide images were collected, and the original database was extended by data enhancement to strengthen generalization ability of models. Subsequently, the CBAM-U-net was constructed by introducing spatial attention module and channel attention module for each down-sampling process in U-net. Meanwhile, U-net, FCN and DeepLabv3 + are used as comparison models. Finally, 6 evaluation metrics were used to comprehensively assess the ability of models for landslide identification and segmentation. The results show that CBAM-U-net exhibited better recognition and segmentation accuracies compared to other models, with optimal values of average row correct, dice coefficient, global correct, IoU and mean IoU of 98.3, 0.877, 95, 88.5 and 90.2, respectively. U-net, DeepLab V3 + , and FCN tend to confuse bare ground and roads with landslides. In contrast, CBAM-U-net has stronger ability of feature learning, feature representation, feature refinement and adaptation.The proposed method can improve the problems of semantic gap and spatial information loss in U-net, and has better accuracy and robustness in recognizing and segmenting high-precision landslide images, which can provide certain reference value for the research of rapid landslide recognition and detection.
引用
收藏
页码:5487 / 5498
页数:12
相关论文
共 4 条
  • [1] A High-Precision Water Body Extraction Method Based on Improved Lightweight U-Net
    An, Shihao
    Rui, Xiaoping
    REMOTE SENSING, 2022, 14 (17)
  • [2] High-precision intelligent identification method of truck overload based on TOI-Net
    Liang, Jian
    Kang, Jiehu
    Zhao, Zongyang
    Wu, Bin
    Wang, Xuesen
    Yi Qi Yi Biao Xue Bao/Chinese Journal of Scientific Instrument, 2024, 45 (05): : 319 - 328
  • [3] MT-U2Net: Lightweight detection network for high-precision magnetic tile surface defect localization
    Jiang, Chenzhe
    Zhang, Xudong
    Xu, Banglian
    Zheng, Qinghe
    Li, Zhengtao
    Zhang, Leihong
    Zhang, Dawei
    MATERIALS TODAY COMMUNICATIONS, 2024, 41
  • [4] Tassel-YOLO: A New High-Precision and Real-Time Method for Maize Tassel Detection and Counting Based on UAV Aerial Images
    Pu, Hongli
    Chen, Xian
    Yang, Yiyu
    Tang, Rong
    Luo, Jinwen
    Wang, Yuchao
    Mu, Jiong
    DRONES, 2023, 7 (08)