基于SE-YOLOv5模型皮带异物检测算法研究

    Research on belt foreign object detection algorithm based on SE-YOLOv5 model

    • 摘要: 以某煤矿1305智能工作面皮带异物识别为工程背景,为了解决井下皮带运输机因废弃锚杆、大块煤等异物而导致的皮带撕裂甚至损伤停机等问题,亟需开展皮带异物智能检测研究。本文提出了SE-YOLOv5皮带异物智能检测方法,该方法以YOLOv5目标检测技术为基础模型,加入SE通道注意力机制进行优化,并对学习率、图像输入批大小、权重衰减等模型参数调整,对构建的数据集进行训练检测,将检测结果与Faster-RCNN、YOLOv3、CenterNet、YOLOv5等模型进行对比。研究结果表明:SE-YOLOv5模型预测结果得到较大提升,对锚杆的预测精度达0.98,对大块煤的预测精度达0.88,召回率(Recall)为0.91,各个检测目标平均精度的平均值(mAP)为0.912,单张识别速度为0.037 s;此外,对处于低照度、高粉尘浓度等环境下的数据集仍有较高的识别率,说明SE-YOLOv5模型检测精度高、速度快、鲁棒性强,可以满足复杂环境下皮带异物识别的要求。煤矿皮带异物检测是煤矿安全和生产效率的关键组成部分,采用现代化的检测技术有助于提高系统生产效率,保障设备安全,降低维护成本。

       

      Abstract: Based on the engineering background of belt foreign object recognition of 1305 intelligent working face in a coal mine, in order to solve the problems of belt tearing and even damage and shutdown caused by foreign objects such as waste bolts and large pieces of coal in underground belt conveyors, it is urgent to carry out research on belt foreign object detection. The SE-YOLOv5 belt foreign object detection method is proposed, which is based on the YOLOv5 object detection technology, and the SE channel attention mechanism is added to optimize the model parameters, and the model parameters such as learning rate, image input batch size, and weight attenuation are adjusted, and the constructed dataset is trained and detected, and the detection results are compared with Faster-RCNN, YOLOv3, CenterNet, YOLOv5 and other models. The research results show that the prediction results of the SE-YOLOv5 model are greatly improved, with the prediction accuracy of the anchor rod reaching 0.98, the prediction accuracy of the large coal being 0.88, the recall rate (Recall) being 0.91, the average accuracy of each detection target (mAP) being 0.912, and the single recognition speed being 0.037s. It shows that the optimized SE-YOLOv5 model has high detection accuracy, fast speed and strong robustness, which can meet the requirements of belt foreign object recognition in complex environments. In general, the belts foreign object detection in coal mine is a key component of coal mine safety and production efficiency, and the use of modern detection technology can help improve system production efficiency, ensure equipment safety, and reduce maintenance costs.

       

    /

    返回文章
    返回