改进Bot-SORT的边坡落石监测方法  

Improved Bot-SORT for Slope Rockfall Monitoring

作　　者：王晓青 阎吉 张德育[1] WANG Xiaoqing;YAN Ji;ZHANG Deyu(Shenyang Ligong University,Shenyang 110159,China)

机构地区：[1]沈阳理工大学信息科学与工程学院,沈阳110159

出　　处：《沈阳理工大学学报》2024年第4期19-26,共8页Journal of Shenyang Ligong University

基　　金：辽宁省科学技术基金项目(2022JH1/10800085)。

摘　　要：针对边坡落石监测中存在的目标尺寸小、石块与背景特征差距小、落石目标运动速度快等问题,提出一种基于检测的改进Bot-SORT多目标跟踪算法。在检测部分对YOLOv7模型进行改进,引入注意力机制,提升模型对石块特征的提取能力,并使用归一化高斯Wasserstein距离作为真值框与预测框的距离度量方式,降低模型对小目标的漏检率;在跟踪部分引入GIoU距离匹配方式,有效跟踪快速运动的落石。通过实景拍摄及Unity仿真方式建立训练及测试数据集,消融实验和对比实验结果表明,本文改进算法能够有效提高落石的检测率和跟踪精度。A detection based improved Bot-SORT multi-objective tracking algorithm is proposed to address the issues of small target size,small difference between rock and background features,and fast movement speed of rockfall targets in slope rockfall monitoring.In the detection section,the YOLOv7 model is improved by introducing attention mechanism to enhance the model’s ability to extract stone features.Normalized Gaussian Wasserstein distance is used as a metric for the ground truth box and target bounding box to reduce the model’s missed detection rate for small targets.The GIoU distance matching method is introduced in the tracking section to effectively track fast falling rocks.The training and testing dataset is established through live shooting and Unity simula-tion.The results of ablation experiments and comparative experiments show that the improved algo-rithm in this paper can effectively improve the detection rate and tracking accuracy of falling rocks.

关 键 词：多目标跟踪 落石监测 YOLOv7 Bot-SORT 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术]