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

Abstract

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.