基于改进YOLOv5的安全帽佩戴检测

摘要：

针对YOLOv5无法通过权重进行聚焦,产生更具有分辨性的特征,从而降低安全帽检测准确性的问题,使用注意力模块,并分别研究压缩激励层(Squeeze and Excitation Layer,SEL)和高效通道注意力(Efficient Channel Attention,ECA)模块.针对YOLOv5去除冗余框时采用的非极大值抑制(Non Maximum Suppression,NMS)在物体高度重叠时仅保留同类最高置信度预测框的问题,使用Soft-NMS算法保留更多的预测框,并进一步使用加权非极大值抑制(Weighted Non Maxi-mum Suppression,WNMS)融合多次预测框信息提升预测框准确性;针对下采样带来的信息丢失问题,使用Focus模块提升检测效果;综合各个模块得到最优的FESW-YOLO算法.该算法在安全帽数据集上的mAP@0.5、mAP@0.5∶0.95相较于YOLOv5分别提高了 2.1个百分点、1.2个百分点,提升了安全帽监管准确性.

关键词：目标检测、安全帽监测、卷积网络、深度学习

分类号：TP391.4(计算技术、计算机技术)

资助基金：国网浙江省电力有限公司双创项目B711JZ200009

在线出版日期：2022-06-29（万方平台首次上网日期，不代表论文的发表时间）

页数：共6页

页码：104-108,126