高切,李登华,丁勇. 基于M-DBT框架的岩质边坡落石跟踪算法研究[J]. 水利水运工程学报,2024.. doi: 10.12170/20230606004
引用本文: 高切,李登华,丁勇. 基于M-DBT框架的岩质边坡落石跟踪算法研究[J]. 水利水运工程学报,2024.. doi: 10.12170/20230606004
(GAO Qie, LI Denghua, DING Yong. A study on the rock block tracking algorithm that utilizes the M-DBT framework[J]. Hydro-Science and Engineering, 2024(in Chinese)). doi: 10.12170/20230606004
Citation: (GAO Qie, LI Denghua, DING Yong. A study on the rock block tracking algorithm that utilizes the M-DBT framework[J]. Hydro-Science and Engineering, 2024(in Chinese)). doi: 10.12170/20230606004

基于M-DBT框架的岩质边坡落石跟踪算法研究

A study on the rock block tracking algorithm that utilizes the M-DBT framework

  • 摘要: 边坡落石是一种常见的自然灾害,构建高效准确的落石监测方法对边坡安全监控十分重要。传统基于视频图像的落石监测方法缺乏对落石运动的关注。针对该问题,选取YOLOv5s和DeepSort模型,基于DBT框架,从框架、算法两个层面对模型进行改进,提出了基于M-DBT框架的落石跟踪算法。框架层面,基于Background Matting算法搭建了运动检测器,并融入原有的DBT框架中,依据运动检测结果动态调用落石跟踪算法,提出了M-DBT目标跟踪算法框架,提高了设备算力的使用效率。算法层面,基于CBAM注意力机制、ASPP空间金字塔池化、SIoU损失函数和Swin Transformer网络集成优化了YOLOv5s和DeepSort算法,提高了算法对落石特征的提取能力、预测框的准确率及检测目标ID分配精度。使用自建目标检测和图像分类数据集进行模型训练与试验,试验结果表明:基于M-DBT框架的落石跟踪算法与基于DBT框架的算法相比,能有效减少算力资源消耗(最高可减少21.5%);改进后的YOLOv5s模型检测精度提高了8.4%,岩块检测准确率最高达87%;改进后DeepSort模型的跟踪精度提升了8.9%,跟踪准确率最高达84%。所提落石跟踪算法能够实现边坡落石的检测与跟踪,具有较强的实用价值。

     

    Abstract: Landslide rockfall poses a significant threat as a common natural disaster, necessitating the development of efficient and accurate monitoring methods for slope safety. Conventional rockfall monitoring techniques based on video imagery often overlook the dynamic movement of fallen rocks. To address this issue, this study focuses on enhancing the YOLOv5s and DeepSort models using the DBT (Dynamic Background Template) framework, leading to the proposal of a rockfall tracking algorithm based on the M-DBT (Modified DBT) framework. At the framework level, an innovative motion detector is constructed by integrating the Background Matting algorithm into the original DBT framework. This integration forms the foundation of the M-DBT target tracking algorithm framework, which dynamically invokes the rockfall tracking algorithm based on motion detection results. The computational efficiency of the monitoring device is significantly improved. At the algorithm level, the YOLOv5s and DeepSort algorithms are optimized by incorporating the CBAM (Convolutional Block Attention Module) attention mechanism, ASPP (Atrous Spatial Pyramid Pooling) spatial pyramid pooling, SIoU (Scale-Invariant Intersection over Union) loss function, and Swin Transformer network. These enhancements augment the algorithms' capacity to extract rockfall features, enhance the accuracy of prediction boxes, and improve the precision of ID allocation for detected targets. The model is trained and evaluated using custom-built target detection and image classification datasets. Experimental results demonstrate the effectiveness of the rockfall tracking algorithm based on the M-DBT framework in significantly reducing computational resource consumption by up to 21.5% compared to the DBT-based algorithm. The improved YOLOv5s model exhibits an impressive 8.4% increase in detection accuracy, with the highest achieved rock detection accuracy reaching 87%. Similarly, the enhanced DeepSort model achieves an 8.9% improvement in tracking accuracy, with the highest attained tracking accuracy reaching 84%. The proposed rockfall tracking algorithm showcases strong practical value by enabling the detection and tracking of landslide rockfalls.

     

/

返回文章
返回