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

RT-DETR-Based Safety Harness Detection for Elevated Operations in Coal Mines

  • 摘要: 为了进一步提高针对登高作业安全绳穿戴状态的检测准确率,保障煤矿井下工作人员生产安全,提出了一种基于RT-DETR的煤矿井下人员登高作业护具穿戴检测模型。首先,将RT-DETR中的原主干网络ResNet替换为轻量级网络ShuffleNetv2,在明显减少模型参数量同时基本保证了检测准确率,提升了模型检测速度;其次,结合Focaler-IoU和MPDIoU构造Focaler-MPDIoU损失函数,解决了GIoU损失函数会忽略两个垂直方向上的几何偏差的问题,降低了漏检率;再使用量化感知训练对模型进行微调,进一步优化模型性能。结果表明:方法检测准确率达97.2%,权重文件大小为17.9MB,与原模型相比,权重文件大小减少了88.9%,在保证检测精度的前提下,有效提升了检测速度,与目前主流的同类目标检测模型相比,具有较高的性能优势和应用价值。

     

    Abstract: To further improve the detection accuracy of safety harness wearing status during high-altitude operations and ensure the safety of underground coal mine personnel, a RT-DETR-based model for detecting safety gear compliance during elevated operations in coal mines. First, the original backbone network ResNet in RT-DETR is replaced with the lightweight network ShuffleNetv2, significantly reducing the model parameters while largely maintaining detection accuracy and improving inference speed. Second, a Focaler-MPDIoU loss function is constructed by integrating Focaler-IoU and MPDIoU, which addresses the issue of the GIoU loss function ignoring geometric deviations in two perpendicular directions and reduces the missed detection rate. Furthermore, quantization-aware training is applied for fine-tuning to further optimize model performance. The results show that the proposed method achieves a detection accuracy of 97.2%, with a weight file size of 17.9 MB. Compared with the original model, the weight file size is reduced by 88.9%. While ensuring detection accuracy, the inference speed is effectively improved. In comparison with current mainstream target detection models, the proposed model demonstrates superior performance and practical applicability.

     

/

返回文章
返回