高级检索

一种DSP-YOLO煤矿带式输送机人员不安全状态智能检测算法

A DSP-YOLO intelligent detection algorithm of unsafe personnel state for coal mine belt conveyor

  • 摘要: 为解决煤矿带式输送机周遭人员不安全状态检测时因尺度变化和背景遮挡导致的准确率低的问题,提出一种DSP-YOLO带式输送机人员不安全状态智能检测算法。首先,针对原始数据多样性不足和模型泛化能力有限的问题,将传统数据增强与GridMask数据增强策略相结合,以提升模型对复杂背景中遮挡目标和小目标检测的鲁棒性。其次,针对人员不安全状态的检测效率与准确性需求,在主干网络中引入感受野扩展卷积(Shift-Wise Conv, SWC)模块,以扩展模型感受野并增强主干网络的全局特征提取能力;同时,设计多尺度特征融合(Dynamic Multi-Scale Fusion and Triple Feature Encoder, DyMSF-TFE)模块,对多尺度特征进行动态融合,充分利用细节特征和多尺度特征,以提高模型对多尺度目标和复杂背景的适应性;并且,采用PIoUv2损失函数,以缓解遮挡目标和小目标引起的漏检和误检问题。此外,为进一步提升DSP-YOLO在实际场景中的检测性能,使用实验室数据训练模型并进行迁移应用。结果表明,结合传统数据增强和GridMask的数据集较原始数据集的精确度、mAP0.5和mAP0.5:0.95分别增加了3.4%、1.9%和6.6%;DSP-YOLO的精确度、召回率、mAP0.5和mAP0.5:0.95分别达到94.8%、89.4%、93.2%和70.5%,较基线模型分别提升了2.3%、1.1%、0.9%和2.3%;针对现场数据,采用模型迁移应用策略可使检测精确度达到99.1%。研究结果证明了所提算法能够实现煤矿带式输送机周遭人员不安全状态的准确智能检测。

     

    Abstract: In order to solve the low detection precision issue of unsafe personnel state around coal mine belt conveyor due to variable scale and background occlusion, a novel DSP-YOLO intelligent fast detection algorithm is proposed. Firstly, to address the problem of insufficient diversity of original data and limited model generalization ability, the traditional data augmentation is improved with the GridMask data augmentation strategy to increase the robustness of the detection algorithm for the occluded or small targets in a complex background. Secondly, to address the requirements of the detection precision and efficiency of unsafe personnel state, the Shift-wise convolution is introduced into the backbone network to extend the receptive field and enhance the global feature extraction ability. Meanwhile, the dynamic multi-scale feature fusion module is designed, making full use of the detail features and multi-scale features to improve the adaptability of the detection algorithm for the multi-scale targets and a complex background. And the PIoUv2 loss function is adopted to alleviate the problem of missing detection and false detection caused by the occluded targets and small targets. In addition, to further improve the detection performance of DSP-YOLO in an actual scene, the detection model is trained using the laboratory data and transferred it for the on-site data. The results show that the precision, mAP0.5 and mAP0.5:0.95 on the data set combined with the traditional data augmentation and the GridMask strategy increase by 3.4%, 1.9% and 6.6%, respectively, compared to the original data set; the precision, recall, mAP0.5 and mAP0.5:0.95 of the DSP-YOLO reach 94.8%, 89.4%, 93.2% and 70.5%, respectively, with an increase of 2.3%, 1.1%, 0.9% and 2.3% compared to the baseline model; for the on-site data, the detection precision reaches 99.1% by using the transfer application strategy. The research results show that the proposed new algorithm is capable of achieving the accurate and intelligent detection of unsafe personnel state around the coal mine belt conveyor.

     

/

返回文章
返回