基于YOLO-DIS露天矿车辆目标检测算法

Based on the YOLO-DIS Open-Pit Mine Vehicle Target Detection Algorithm

  • 摘要: 针对露天矿山作业区环境中存在的复杂背景干扰、障碍物遮挡以及扬尘影响小目标漏检等问题,提出了一种YOLO11的矿山车辆目标检测改进算法,记作YOLO-DIS。首先通过引入iAFF迭代注意力来改进传统的C3k2模块,设计一种C3k2_iAFF结构,来强化复杂情况下的特征提取能力。其次,采用轻量化动态上采样Dysample代替了传统的上采样方法,有效的弥补了因固定规则导致被遮挡目标的边缘特征恢复不准确的缺陷。最后,采用SlideLoss损失函数实现动态调整来应对样本分布不均的问题。实验结果表明,在自制数据集上的检测精度为81.5%,相较于原始模型YOLO11 mAP50%提升了4%。为了验证改进算法的泛化性,算法在KITTI数据集上进行了验证,mAP50%从0.911提升到了0.926,表现出了更优的检测精度,体现了算法的实用性和科学性。

     

    Abstract: To address issues such as complex background interference, obstacle occlusion, and missed detection of small targets caused by dust in the open-pit mine operation area environment, an improved YOLO11-based target detection algorithm for mine vehicles, denoted as YOLO-DIS, is proposed. Firstly, the iterative attention mechanism (iAFF) is introduced to improve the traditional C3k2 module, and a C3k2_iAFF structure is designed to enhance the feature extraction capability under complex conditions. Secondly, the lightweight dynamic upsampling method (Dysample) is adopted to replace the traditional upsampling method, which effectively compensates for the defect of inaccurate edge feature restoration of occluded targets caused by fixed rules. Finally, the SlideLoss function is used to realize dynamic adjustment to tackle the problem of uneven sample distribution. Experimental results show that the detection accuracy of the proposed algorithm on the self-made dataset reaches 81.5%, which is 4% higher than the mAP50% of the original YOLO11 model. To verify the generalization ability of the improved algorithm, validation experiments are conducted on the KITTI dataset, where the mAP50% increases from 0.911 to 0.926. The algorithm demonstrates better detection accuracy, indicating its practicality and scientificity.

     

/

返回文章
返回