基于改进RT−DETR的煤矿井下人员登高作业护具穿戴检测

Improved RT-DETR-based detection of personal protective equipment wearing in underground coal mine working-at-height operations

  • 摘要: 现有基于YOLO系列模型的穿戴检测方法在复杂背景、遮挡和小目标检测条件下易出现漏检或误检,基于RT−DETR的模型应用于煤矿井下登高作业护具穿戴检测时存在参数量过大,易受人体姿态变化、肢体遮挡和光照不均影响而导致边界框定位不准确等问题。针对上述问题,提出了一种基于改进RT−DETR的煤矿井下人员登高作业护具穿戴检测模型。将RT−DETR中的原主干网络ResNet替换为轻量级网络ShuffleNetv2,在明显减少模型参数量的同时基本保证了检测准确率,使模型能够部署于资源受限的井下边缘设备;构造了Focaler−MPDIoU损失函数,结合对边界框角点位置偏差的精细约束能力和自适应聚焦机制,增强对遮挡、姿态多变及小目标等困难样本的定位能力,提升目标检测精度与框回归质量;使用量化感知训练进一步压缩模型体积,提升模型在井下边缘设备上部署的适应性。实验结果表明:改进RT−DETR模型在自建数据集上的准确率达97.2%,参数量为14.5×106个,与原模型相比缩减了65.4%,极大提升了模型的轻量化水平;与主流轻量级模型YOLOv5n,YOLOv8n,YOLOv11n相比,mAP@50分别提高了4.1%,4.0%,3.2%,在检测精度上具有明显优势。

     

    Abstract: Existing YOLO-series-based detection methods for personal protective equipment wearing often suffer from missed detections and false detections in complex backgrounds, occlusions, and small object scenarios. When RT-DETR-based models are applied to personal protective equipment wearing detection in underground coal mine working-at-height operations, they face issues such as excessive model parameters and inaccurate bounding box localization caused by human pose variations, limb occlusion, and uneven illumination. To address these problems, an improved RT-DETR-based detection model for personal protective equipment wearing in underground coal mine working-at-height operations was proposed. The ResNet backbone in RT-DETR was replaced with the lightweight ShuffleNetv2 network, significantly reducing the number of model parameters while maintaining detection accuracy, enabling deployment on resource-constrained underground edge devices. A Focaler-MPDIoU loss function was introduced, combining fine-grained constraints on bounding box corner deviations and an adaptive focusing mechanism, enhancing detection accuracy and bounding box regression quality, and improving localization performance for occluded, pose-varying, and small targets. Quantization-aware training was further adopted to compress the model size and improve its adaptability for deployment on underground edge devices. Experimental results showed that the proposed model achieved an accuracy of 97.2% on a self-built dataset, with a parameter size of 14.5×106, representing a 65.4% reduction compared with the original model and greatly improving the lightweight level of the model. Compared with the mainstream lightweight models YOLOv5n, YOLOv8n, and YOLOv11n, mAP@50 was improved by 4.1%, 4.0%, and 3.2%, respectively, demonstrating superior detection accuracy.

     

/

返回文章
返回