• 论文
主办单位:煤炭科学研究总院有限公司、中国煤炭学会学术期刊工作委员会
融合坐标注意力与多尺度的轻量级安全帽佩戴检测
  • 80
  • 作者

    李忠飞冯仕咏郭骏张云鹤徐飞翔

  • 单位

    内蒙古电投能源股份有限公司北露天煤矿中国矿业大学信息与控制工程学院 北京和利时数字技术有限公司

  • 摘要
    针对现有煤矿工人安全帽佩戴检测算法存在检测精度与速度难以取得较好平衡的问题,以YOLOv4模型为基础,提出了一种融合坐标注意力与多尺度的轻量级模型M-YOLO,并将其用于安全帽佩戴检测。该模型使用融入混洗坐标注意力模块的轻量化特征提取网络S-MobileNetV2替换YOLOv4的特征提取网络CSPDarknet53,在减少相关参数量的前提下,有效改善了特征之间的联系;将原有空间金字塔池化结构中的并行连接方式改为串行连接,有效提高了计算效率;对特征融合网络进行改进,引入具有高分辨率、多细节纹理信息的浅层特征,以有效加强对检测目标特征的提取,并将原有Neck结构中的部分卷积修改为深度可分离卷积,在保证检测精度的前提下进一步降低了模型的参数量和计算量。实验结果表明,与YOLOv4模型相比,M-YOLO模型的平均精度均值仅降低了0.84%,但计算量、参数量、模型大小分别减小了74.5%,72.8%,81.6%,检测速度提高了53.4%;相较于其他模型,M-YOLO模型在准确率和实时性方面取得了良好的平衡,满足在智能视频监控终端上嵌入式加载和部署的需求。
  • 关键词

    目标检测安全帽佩戴检测坐标注意力模块轻量化多尺度特征融合

  • 文章目录
    0引言
    1 M-YOLO模型原理
    1.1 M-YOLO结构
    1.2 S-MobileNetV2特征提取网络
    1.3 SPP结构改进
    1.4 特征融合网络重构
    2 实验结果与分析
    2.1 基础MobileNetV2特征提取能力实验
    2.2 不同SCA模块位置实验
    2.3 消融实验
    2.4 模型对比实验
    2.5 实际场景检测效果对比实验
    3 结论
相关问题

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

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