• 论文
主办单位:煤炭科学研究总院有限公司、中国煤炭学会学术期刊工作委员会
基于改进YOLOv5s的矿工排队检测方法
  • 93
  • 作者

    郝明月闵冰冰张新建赵作鹏吴晨王欣

  • 单位

    河南龙宇能源股份有限公司陈四楼煤矿中国矿业大学计算机科学与技术学院

  • 摘要
    传统的目标检测算法识别矿工排队异常行为时需要人工提取特征,检测时间长、检测精度低;基于卷积神经网络的目标检测算法在检测速度和精度上有所提升,但在遮挡、昏暗和光照不均等场景下的检测效果难以保障。针对上述问题,提出了一种改进YOLOv5s(HPI-YOLOv5s)模型,并将其用于矿工排队检测。HPI-YOLOv5s模型在YOLOv5s模型的基础上对路径聚合网络(PANet)进行改进,通过删除单个输入边节点、增加双向交叉路径,构建了一种双向交叉特征金字塔网络(BCrFPN)进行多尺度特征融合。鉴于手动设置阈值的标签分配策略鲁棒性不高,在自适应训练样本选择(ATSS)动态设置阈值的基础上,提出动态标签分配策略(ATSS_PLUS),更合理地评估候选样本的质量,动态设定每个真实目标的阈值,具有更高的检测精度和鲁棒性。通过半平面交法计算人脸框与所划定排队区域的相交面积,并将相交面积和人脸框面积之比与设置的阈值比较来判断矿工是否有序排队。实验结果表明:HPI-YOLOv5s模型比YOLOv5s模型的准确率提高了1.9%,权重大小减少了32%,参数量减少了6.9%,检测速度提高了7.8%,且针对遮挡、昏暗、光照不均的矿井图像,能够更准确地识别矿工排队情况。
  • 关键词

    矿工排队检测人脸检测双向交叉特征金字塔网络特征融合自适应训练样本选择动态标签分配

  • 文章目录
    0 引言
    1 方法原理
    1.1 BCrFPN
    1.2 ATSS_PLUS
    1.3 排队区域提取及检测
    2 实验与分析
    2.1 实验平台
    2.2 数据集及预处理
    2.3 模型性能评估
    2.4 消融实验
    2.5 阈值选择
    2.6 对比实验
    3 结论
  • 引用格式
    郝明月,闵冰冰,张新建等.基于改进YOLOv5s的矿工排队检测方法[J/OL].工矿自动化:1-8[2023-11-17].DOI:10.13272/j.issn.1671-251x.2023030058.
相关问题

主办单位:煤炭科学研究总院有限公司 中国煤炭学会学术期刊工作委员会

©版权所有2015 煤炭科学研究总院有限公司 地址:北京市朝阳区和平里青年沟东路煤炭大厦 邮编:100013
京ICP备05086979号-16  技术支持:云智互联