基于一种改进YOLOv7算法模型的滑坡识别研究——以四川省白格地区为例
Landslide identification based on an improved YOLOv7 model: A case study of the Baige area
-
摘要: 滑坡识别一直是地质灾害领域的一个研究热点,对于抢险应急指挥具有重要意义。针对滑坡检测的漏检误检以及识别精度不高的问题,该文通过对核心网络进行数据融合、增加卷积块注意力模块(convolutional block attention module,CBAM)、更改交并比(intersection over union,IoU)损失函数的优化等方法,提出一种改进的YOLOv7算法模型,实现了对滑坡同时进行目标检测与图像分割的功能,并以贵州省毕节市的滑坡数据集和四川省历史滑坡0.2 m高分辨率航空正射影像为例对模型的有效性进行了验证。研究结果表明,优化后的模型在山体滑坡的检测与分割任务中展现出较好的性能,相比于常规的YOLOv7模型以及Faster RCNN,Mask RCNN等主流模型方法,该模型对山体滑坡的识别更加高效、准确。以四川省白格地区为例,该模型在提高精度的同时可有效提高山体滑坡灾害信息获取的自动化程度。Abstract: Landslide identification has always been a research topic in the study of geological disasters, playing a significant role in emergency rescue and command. To address the limitations in landslide identification, such as missed/false detection, and low identification accuracy, this study proposed an improved YOLOv7 model that enables simultaneous object detection and image segmentation for landslides. The improved model optimized its core network by integrating data, adding the convolutional block attention module (CBAM), and changing the intersection over union (IoU) loss function. Its effectiveness was verified using the landslide dataset of Bijie City, Guizhou Province, and the 0.2 m high-resolution digital orthophoto map (DOM) of historical landslides in Sichuan Province. The results indicate that the improved model performed well in landslide detection and segmentation, achieving more efficient and accurate landslide identification compared to the conventional YOLOv7 model, and other prevailing models like Fast RCNN and Mask RCNN. Taking the Baige area in Sichuan Province as an example, this model can effectively enhance the automation level of landslide disaster information acquisition while improving accuracy.
下载: