水利水电技术（中英文）2024，Vol.55Issue(4)：176-185,10.DOI:10.13928/j.cnki.wrahe.2024.04.016
基于Swin Transformer目标全景分割的三峡库首土质滑坡识别
Identification of soil landslides at the head of the Three Gorges Reservoir based on swin transformer target panoramic segmentation
摘要
Abstract
[Objective]Landslide identification is the key to solve the problem of where the geological disaster hazards are in moun-tainous areas.Artificial intelligence,especially deep learning method,began to be widely used in the field of target recognition,but for landslide hazard recognition in complex environments in multi-vegetation mountainous areas,there are problems such as sin-gle model and poor accuracy.[Methods]Therefore,an intelligent recognition method based on Swin Transformer(Shift Windows Transformer)as backbone network combined with panoramic target segmentation is proposed in this paper to identify soil landslide in the head area of the Three Gorges Reservoir.The 485 soil landslides at the head of the Three Gorges Reservoir are made into a sample set and divided into a training set and a test set.The training set is loaded into the Swin Transformer model for training.The model adopts the self-attention mechanism to extract features from the training set and construct feature maps.Finally,this method can achieve effective differentiation between the landslide target and the background area,and then complete the potential hazards identification.At the same time,it is compared with DeepLab V3 model.[Results]The result show that the Swin Trans-former model is higher than the DeepLab V3 model in recognition accuracy and recognition speed,and the accuracy can reach 83.55％in the experiment at the head of the Three Gorges reservoir,and the prediction time of a single image is 0.18 s.[Con-clusion]The result show that the method can rapidly identify soil landslides in the complex environment of multi-vegetation moun-tainous areas,and can provide a reference for landslide hazard investigation of multi-vegetation mountainous areas.关键词
三峡库首/土质滑坡/Swin Transformer/全景分割/隐患识别/滑坡Key words
the Three Gorges Reservoir Head/soil landslide/Swin Transformer/panoramic segmentation/hazard identification/landslide分类
天文与地球科学引用本文复制引用
邓志勇,黄海峰,李清清,周红,张瑞,柳青,董志鸿..基于Swin Transformer目标全景分割的三峡库首土质滑坡识别[J].水利水电技术（中英文）,2024,55(4):176-185,10.基金项目
国家自然科学基金(U21A2031,42007237,42107489) （U21A2031,42007237,42107489）
三峡库区地质灾害教育部重点实验室开放基金(2020KDZ09) （2020KDZ09）
