Skip to main content

Advertisement

Log in

Methods for landslide detection based on lightweight YOLOv4 convolutional neural network

  • Research Article
  • Published:
Earth Science Informatics Aims and scope Submit manuscript

Abstract

The rapid and accurate positioning of the landslides through remote sensing data plays an important role in post-disaster emergency rescue. This paper was proposed a new algorithm for landslide detection in the plateau environment. The YOLOv4 was used as the basic framework, and the MobileNetv3 model was utilized as the feature extraction network to replace the backbone neural network CSPdarknet53 which was to improve the efficiency of landslide detection. By applying depth separable convolution, the parameters of the model are decreasing significantly. To further improve the accuracy of landslide detection, the coordinate attention mechanism was introduced in the bottleneck. 3070 landslide images in the Linzhi area from 2010 to 2019 were obtained through Google Earth to train and test the model. On this basis, we compared the detection speed and accuracy of other single-stage and two-stage target detection algorithms in landslide detection. Moreover, the performances of the model were analyzed under the different attention mechanisms. The results show that our model can reduce the number of parameters by 83.59% compared with the YOLOv4 model. The accuracy of landslide detection by the model is improved to 91.2%, and the detection rate reaches 35f/s. It means that the model proposed in this study would provide useful information and rapid detection for hazard assessment and emergency rescue.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6.
Fig. 7
Fig. 8

Similar content being viewed by others

References

  • Asselen SV, Seijmonsbergen AC (2006)Expert-driven semi-automated geomorphological mapping for a mountainous area using a laser DTM. Geomorphology 78(3/4):309–320

    Article  Google Scholar 

  • Bochkovskiy A, Wang CY, Liao H (2020) Yolov4: Optimal speed and accuracy of object detection, arXiv preprint arXiv:10934

  • Bui DT, Tuan TA, Klempe H, Klempe H, Pradhan B, Revahaug I (2016) Spatial prediction models for shallow landslide hazards: a comparative assessment of the efficacy of support vector machines, artificial neural networks, kernel logistic regression, and logistic model tree. Landslides 13(2):361–378

    Article  Google Scholar 

  • Cervi F, Berti M, Borgatti L, Ronchetti F, Manenti F, Corsini A (2010) Comparing predictive capability of statistical and deterministic methods for landslide susceptibility mapping: a case study in the northern Apennines (Reggio Emilia Province, Italy). Landslides 7(4):433–444

    Article  Google Scholar 

  • Chollet F (2017) Xception: deep learning with Depthwise separable convolutions, IEEE Conference on Computer Vision Pattern Recognition

  • Danneels G, Pirard E, Havenith HB (2007) Automatic landslide detection from remote sensing images using supervised classification methods, IEEE International Geoscience & Remote Sensing Symposium

  • Galli M, Ardizzone F, Cardinali M, Guzzetti F, Reichenbach P (2008) Comparing landslide inventory maps. Geomorphology 94(3–4):268–289

    Article  Google Scholar 

  • Ghorbanzadeh O, Blaschke T, Gholamnia K, Meena S, Tiede D, Aryal J (2019) Evaluation of different machine learning methods and deep-learning convolutional neural networks for landslide detection. Remote Sens 11(2)

  • Guo J, Li Y, Lin W, Chen Y, Li J (2018) Network decoupling: from regular to Depthwise separable convolutions, 29the British machine vision conference (BMVC 2018)

  • Hacıefendioğlu K, Demir G, Başağa HB (2021) Landslide detection using visualization techniques for deep convolutional neural network models. Nat Hazards 109:1–22

    Article  Google Scholar 

  • Hou Q, Zhou D, Feng J (2021) Coordinate attention for efficient mobile network design, proceedings of the IEEE/CVF conference on computer vision and pattern recognition, pp 13713–13722

  • Howard AG, Zhu M, Chen B, Kalenichenko D,Wang W, Weyand T, Andreetto M, Adam H (2017) MobileNets: efficient convolutional neural networks for Mobile vision applications. arXiv preprint arXiv:1704.04861

  • Howard A, Sandler M, Chu G, Chen L, Tan M, Wang W, Zhu Y, Pang R, Vasudevan V (2019) Searching for mobilenetv3, proceedings of the IEEE/CVF international conference on computer vision, pp 1314–1324

  • Hu J, Shen L, Albanie S, Sun G, Vedaldi A (2018) Gather-excite: Exploiting feature context in convolutional neural networks. arXiv preprint arXiv:1810.12348

  • Jie H, Li S, Gang S, Albanie S (2017)Squeeze-and-excitation networks. IEEE Transactions on Pattern Analysis Machine Intelligence, pp 99

  • Loshchilov I, Hutter F (2017) SGDR: stochastic gradient descent with warm restarts. arXiv: learning

  • Martha TR, Kerle N, Jetten V, van Westen CJ, Kumar KV (2010) Characterising spectral, spatial and morphometric properties of landslides for semi-automatic detection using object-oriented methods. Geomorphology 116(1–2):24–36

    Article  Google Scholar 

  • McDERMID GJ, Franklin S (1995) Remote sensing and geomorphometric discrimination of slope processes. Zeitschrift für Geomorphologie. Supplementband 101:165–185

    Google Scholar 

  • Redmon J, Farhadi A (2018) YOLOv3: an incremental improvement. ArXiv, abs/1804.02767

  • Redmon J, Divvala S, Girshick R, Farhadi A (2016) You only look once: unified, Real-Time Object Detection. In Proceedings of the IEEE conference on computer vision and pattern recognition, pp 779–788

  • Ren SQ, He KM, Girshick R et al (2015) Faster R-CNN: towards real-time object detection with region proposal networks. Advances in neural information processing systems 28 (nips 2015):28

  • Ren S, He K, Girshick R, Sun J (2017) Faster R-CNN: towards real-time object detection with region proposal networks. IEEE Trans Pattern Anal Mach Intell 39(6):1137–1149

    Article  Google Scholar 

  • Saha AK, Gupta RP, Sarkar I, Arora MK, Csaplovics E (2005) An approach for GIS-based statistical landslide susceptibility zonation—with a case study in the Himalayas. Landslides 2(1):61–69

    Article  Google Scholar 

  • Sandler M, Howard A, Zhu M, Zhmoginov A, Chen LC (2018) MobileNetV2: inverted residuals and linear bottlenecks, 2018 IEEE/CVF conference on computer vision and pattern recognition (CVPR), pp 4510–4520

  • Tsotsos JK (1990) Analyzing vision at the complexity level. Behav Brain Sci 13(3):423–445

    Article  Google Scholar 

  • Wang CY, Liao H, Wu YH, Chen PY, Hsieh JW, Yeh IH (2020a) CSPNet: A New Backbone that can Enhance Learning Capability of CNN, 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW), pp 390–391

  • Wang Q, Wu B, Zhu P, Li P, Zuo W, Hu Q (2020b) ECA-net: Efficient Channel attention for deep convolutional neural networks, 2020 IEEE/CVF conference on computer vision and pattern recognition (CVPR)

  • Woo S, Park J, Lee JY, Kweon IS (2018) CBAM: convolutional block attention module. Proceedings of the European conference on computer vision (ECCV), pp 3–19

  • Xu ZG, Chen YM, Yang F, Chu T, Zhou H (2020) A Postearthquake multiple scene recognition model based on classical SSD method and transfer learning. ISPRS Int J Geo Inf 9(4):238

  • Zhou H, Yong W, Miao Y (2018) A method of CNN traffic classification based on Sppnet, 2018 14th international conference on computational intelligence and security (CIS)

Download references

Acknowledgments

The research is supported by the National Key R&D Program of China (2016YFC0401600 and 2017YFC0404900).

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Junjie Li.

Additional information

Communicated by: H. Babaie

Publisher’s note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Li, B., Li, J. Methods for landslide detection based on lightweight YOLOv4 convolutional neural network. Earth Sci Inform 15, 765–775 (2022). https://doi.org/10.1007/s12145-022-00764-0

Download citation

  • Received:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s12145-022-00764-0

Keywords

Navigation