基于注意力机制的矿工危险行为检测算法研究

    Study on detection algorithm for miner dangerous behavior based on attention mechanism

    • 摘要: 为了有效地检测和识别煤矿井下工作人员的危险行为,防止安全事故的发生,针对煤炭井下背景复杂、尺度变化大等问题,提出了一种基于注意力机制和深度学习的矿工危险动作检测算法。在YOLOv3模型的基础上,设计一种特征提取能力更强、体积更小的轻量化特征提取网络;针对原始的YOLOv3算法在小目标的检测性能较差这一问题,提出了一种基于注意力机制的特征融合模块来优化小目标的漏检和误检问题。为了评估模型的性能,采集了10 000张煤矿井下图片用于训练和测试,所提出的算法的mAP为83.1%,优于目前常用的目标检测算法;此外,算法测试速度为769 fps,是其他轻量化目标检测算法的6.6倍。试验结果证明,提出的危险行为检测算法可以应用到实际的生产环境中。

       

      Abstract: In order to effectively detect and identify the dangerous behaviors of underground workers in coal mines and prevent the occurrence of safety accidents, a detection algorithm for dangerous actions of miners based on attention mechanism and deep learning is proposed for the problems of complex background and large scale changes in coal mines. On the basis of the YOLOv3 model, a lightweight feature extraction network with stronger feature extraction capability and smaller volume is designed; in view of the poor detection performance of the original YOLOv3 algorithm on small targets, an attention-based approach is proposed. The feature fusion module of the mechanism optimizes the missed detection and false detection of small targets. In order to evaluate the performance of the model, 10 000 underground coal mine pictures were collected for training and testing. The mAP of the algorithm proposed in this paper is 83.1%, which is 6.6 times higher than the current commonly used target detection algorithms. In addition, the algorithm test speed is 769 fps, which is faster than other light weight object detection algorithms. The experimental results prove that the dangerous behavior detection algorithm proposed in this paper can be applied to the actual production environment.

       

    /

    返回文章
    返回