摘要
针对在智能监控中安全帽佩戴检测准确率低和检测速率慢的问题,提出一种基于改进YOLOv3(You Only Look Once)的安全帽佩戴检测算法YOLOv3-WH。在YOLOv3算法的基础上改进网络结构,增大输入图像的尺度,使用深度可分离卷积结构替换Darknet-53传统卷积,减少特征的丢失,缩减模型参数,提升检测速率;使用多尺度特征检测,增加浅层检测尺度,添加4倍上采样特征融合结构,提高安全帽佩戴检测准确率;优化K-Means聚类算法,获取安全帽佩戴检测的先验框(anchor box),按照预测尺度大小分配适合的anchor,提升模型训练和检测速率。实验结果表明YOLOv3-WH相比YOLOv3,每秒检测帧数(FPS)提高了64%,检测平均精确度(mAP)提高了6.5%,该算法在提升了安全帽佩戴检测速率的同时提升了检测的准确率,对安全帽佩戴检测具有一定的实用性。
Aiming at the problem of low accuracy and slow detection rate of helmet wearing detection in intelligent moni-toring,a helmet wearing detection algorithm YOLOv3-WH based on improved YOLOv3(You Only Look Once)is pro-posed.The network structure on the basis of the YOLOv3 algorithm is improved,the scale of the input image is increased.Using deep separable convolution structure can replace the Darknet-53 traditional convolution,reduce the loss of features,model parameters,and increase the detection rate.Using multi-scale feature detection can increase the shallow detection scale,add 4 times the up-sampling feature fusion structure and improve the accuracy of helmet wearing detection.Opti-mizing the K-Means clustering algorithm can obtain the anchor box of the helmet wearing detection,according to the pre-dicted scale size allocate suitable anchors,the model training and detection rate are improved.Experimental results show that compared with YOLOv3,YOLOv3-WH has a 64%increase in frame detection per second(FPS),and an average detection accuracy(mAP)has increased by 6.5%.The algorithm improves the detection rate of helmet wearing while improving detection.The accuracy rate is of certain practicality for the detection of helmet wearing.
作者
肖体刚
蔡乐才
高祥
黄洪斌
张超阳
XIAO Tigang;CAI Lecai;GAO Xiang;HUANG Hongbin;ZHANG Chaoyang(School of Automation and Information Engineering,Sichuan University of Science&Engineering,Zigong,Sichuan 643000,China;Sanjiang Artificial Intelligence and Robot Research Institute,Yibin University,Yibin,Sichuan 644000,China)
出处
《计算机工程与应用》
CSCD
北大核心
2021年第12期216-223,共8页
Computer Engineering and Applications
基金
四川省科技厅重点研发项目(2019YFN0104,2016ZGY021)
四川省高校重点实验室开放基金(2018WZY03)。
