现代计算机2022,Vol.28Issue(11) :60-66.DOI:10.3969/j.issn.1007-1423.2022.11.011

轻量化超参数卷积神经网络的安全帽检测算法

Safety Helmet Detection Algorithm Based on Lightweight Hyperparametric Convolutional Neural Network

李青云 李燕 卢峥松 林思伟
现代计算机2022,Vol.28Issue(11) :60-66.DOI:10.3969/j.issn.1007-1423.2022.11.011

轻量化超参数卷积神经网络的安全帽检测算法

Safety Helmet Detection Algorithm Based on Lightweight Hyperparametric Convolutional Neural Network

李青云 1李燕 2卢峥松 1林思伟1
扫码查看

作者信息

  • 1. 南京信息工程大学自动化学院,南京 210044
  • 2. 南京信息工程大学自动化学院,南京 210044;无锡学院物联网学院,无锡 214105
  • 折叠

摘要

目前工地安全帽检测任务中基于深度学习的安全帽识别方法由于底层特征缺少语义与全局信息,从而导致误检与漏检的情况发生.为在保证检测网络具备实时性的情况下,使得网络具有较高的监测性能,本文提出一种轻量化超参数卷积神经网络的安全帽检测算法.该算法以RFB-MobileNet网络为基础,在骨干网络中增加多频谱通道注意力网络(Frequency Channel Attention Networks,FcaNet),增强了网络对于不同尺寸目标特征的提取能力,大幅度提升了算法网络的检测精度,实现了常规施工环境下不同形态大小安全帽的识别.此外,本文将网络中感受野(Receptive Field Block,RFB)模块和FcaNet网络中的常规卷积替换为深度超参数化卷积(Depthwise Over-parameterized Convolutional,DO-Conv),在不增加计算复杂性的情况下,进一步提升模型的检测性能.实验结果表明,本文模型(DO-RFB-Mobilenet)的mAP较RFB-MobileNet的mAP提高了7.08%,检测精度达到74.27%,检测速度达到100帧/秒.

关键词

卷积神经网络/感受野结构/注意力机制/安全帽检测

引用本文复制引用

出版年

2022
现代计算机
中大控股

现代计算机

影响因子:0.292
ISSN:1007-1423
参考文献量3
段落导航相关论文