中国安全科学学报 ›› 2023, Vol. 33 ›› Issue (7): 82-89.doi: 10.16265/j.cnki.issn1003-3033.2023.07.2226

• 安全工程技术 • 上一篇    下一篇

基于改进YOLOv5s的综采工作面人员检测算法

张磊1,2(), 李熙尉1, 燕倩如1, 王浩盛1,**(), 雷伟强1   

  1. 1 山西大同大学 煤炭工程学院,山西 大同 037003
    2 山西大同大学 智能采矿装备产业技术创新研究院(产业学院),山西 大同 037003
  • 收稿日期:2023-02-16 修回日期:2023-05-12 出版日期:2023-07-28
  • 通讯作者:
    ** 王浩盛(1996—),男,山西忻州人,硕士研究生,主要研究方向为智能采矿、煤矿安全等。E-mail:
  • 作者简介:

    张磊 (1984—),男,山西大同人,硕士,副教授,主要从事智能采矿、煤矿地质、煤矿安全等方面的研究。E-mail:

  • 基金资助:
    山西省研究生教育创新项目(2021Y739); 山西大同大学研究生教育创新项目(21CX02); 山西大同大学研究生教育创新项目(21CX37); 山西大同大学2022年度校级揭榜招标项目(2021ZBZX3); 山西大同大学2021年度产学研专项研究项目(2021CXZ2)

Personnel detection algorithm in fully mechanized coal face based on improved YOLOv5s

ZHANG Lei1,2(), LI Xiwei1, YAN Qianru1, WANG Haosheng1,**(), LEI Weiqiang1   

  1. 1 School of Coal Engineering, Shanxi Datong University, Datong Shanxi 037003, China
    2 Intelligent Mining Equipment Industry Technology Innovation Research Institute (Industrial College), Shanxi Datong University, Datong Shanxi 037003, China
  • Received:2023-02-16 Revised:2023-05-12 Published:2023-07-28

摘要:

为了智能监控井工煤矿综采工作面危险区域人员闯入和安全帽佩戴问题,避免监控视频受粉尘干扰、光照不均等因素影响图像检测精度的问题,提出一种基于改进YOLOv5s的目标检测算法(简称YOLOv5s-DPE),并建立相关模型。首先,在颈部网络部分,采用深度可分离卷积(DwConv)替换普通卷积,降低参数量和计算量;然后,引入改进的路径聚合网络(PANet)提升特征提取能力,替换边界框损失函数完全交并比(CIOU)为有效交并比(EIOU),提升检测准确率;最后,选取综采工作面视频中的人员图像进行检测,选取煤矿井下人员闯入和安全帽佩戴监控视频作为检测数据集,并进行训练和验证。结果表明:对比初始YOLOv5s算法模型,YOLOv5s-DPE算法模型的参数量下降14.2%,浮点数计算量下降7.6%,算法网络模型大小下降12.5%,均值平均精度(mAP)@0.5提升到93.7%,mAP@0.5:0.95提升到65.8%,YOLOv5s-DPE模型对小目标检测效果更好,误检漏检等情况有所减少。

关键词: YOLOv5s, 综采工作面, 检测算法, 深度可分离卷积(DwConv), 有效交并比(EIOU), 路径聚合网络(PANet)

Abstract:

In order to intelligently monitor the intrusion of personnel entering dangerous areas and the wearing of safety helmets in the fully mechanized mining face of underground coal mines, an improved object detection algorithm based on YOLOv5s was proposed to solve the problem of dust interference and uneven illumination affecting image detection accuracy in monitoring videos. Firstly, in the neck network section, DwConv was used to replace ordinary convolutions, which reduced parameters and computational complexity. Then, an improved PANet was introduced to improve the feature extraction capability, replacing the bounding box Loss function CIOU (Complete-Intersection Over Union) with EIOU to improve the detection accuracy. Finally, the personnel images in the video of the fully mechanized mining face were selected for detection, and monitoring videos of coal mine underground personnel entering and wearing safety helmets were selected as the detection dataset for training and verification. The results show that, compared with the initial YOLOv5s algorithm network model, the number of parameters of the YOLOv5s-DPE algorithm network model is decreased by 14.2%, the number of floating-point arithmetic calculations is decreased by 7.6%, the size of the algorithm network model is decreased by 12.5%, mAP@0.5 is increased to 93.7%, and mAP@0.5:0.95 is increased to 65.8%. The YOLOv5s-DPE model has better detection performance for small targets, with a reduction in false detections and missed detections.

Key words: YOLOv5s, fully mechanized coal face, detection algorithm, depthwise separable convolution (DwConv), efficient intersection over union (EIOU), path aggregation network (PANet)