基于YOLO−DIS的露天矿复杂环境下车辆障碍物检测

Vehicle obstacle detection in complex open-pit mine environments based on YOLO-DIS

  • 摘要: 针对露天矿作业区遮挡严重、扬尘干扰、图像模糊等复杂环境下车辆障碍物检测存在漏检、误检的问题,在YOLOv11n的基础上提出了一种YOLO−DIS模型,用于露天矿复杂环境下车辆障碍物检测。该模型引入迭代注意力特征融合(iAFF)改进C3k2模块,通过两阶段迭代注意力融合强化了复杂情况下的特征提取能力;采用轻量化动态上采样代替原有的最近邻插值法,通过学习采样点偏移量,根据目标形状及遮挡情况动态调整采样位置,有效弥补了因固定采样规则导致的被遮挡目标边缘特征恢复不准确的缺陷;采用SlideLoss损失函数对不同难度的样本赋予差异化权重,解决了样本分布不均衡问题。实验结果表明:相较于YOLOv11n,YOLO−DIS模型在参数量仅少量增加的情况下,精确率、召回率、mAP@0.5分别提升了4.4%,7.3%,4.0%;与主流目标检测模型相比,YOLO−DIS模型的mAP@0.5最高;YOLO−DIS模型在自制数据集和KITTI数据集上均保持了较好的检测性能,具有良好的泛化性;在遮挡严重、扬尘干扰、图像模糊、小目标检测、背景干扰等场景下,YOLO−DIS模型检测框置信度更高,有效减少了漏检情况。

     

    Abstract: To address the problems of missed and false detections in vehicle obstacle detection under the complex conditions of severe occlusion, dust interference, and image blurring in open-pit mine operational areas, this study proposed a YOLO-DIS model based on YOLOv11n for vehicle obstacle detection in such complex open-pit mine environments. To enhance feature extraction capability in complex situations, the model introduced an Iterative Attention Feature Fusion (iAFF) mechanism to improve the C3k2 module, strengthening feature extraction through two-stage iterative attention fusion. To effectively compensate for the inaccurate recovery of edge features in occluded targets caused by fixed sampling rules, lightweight dynamic upsampling was adopted to replace the original nearest-neighbor interpolation method. This approach dynamically adjusted sampling positions based on target shape and occlusion by learning sampling point offsets. Furthermore, to solve the problem of imbalanced sample distribution, the SlideLoss function was employed to assign differentiated weights to samples of varying difficulty levels. Experimental results demonstrated that: compared to YOLOv11n, the YOLO-DIS model achieved improvements of 4.4%, 7.3%, and 4.0% in precision, recall, and mAP@0.5, respectively, with only a marginal increase in the number of parameters; compared to mainstream object detection models, the YOLO-DIS model achieved the highest mAP@0.5; the YOLO-DIS model maintained good detection performance on both a custom dataset and the KITTI dataset, indicating strong generalization capability. The YOLO-DIS model provided higher confidence levels for detection bounding boxes in scenarios involving severe occlusion, dust interference, image blurring, small target detection, and background interference, effectively reducing missed detections.

     

/

返回文章
返回