基于改进YOLOv7的防护用品穿戴检测
Detection of Wearing Personal Protective Equipment Based on Improved YOLOv7
杨晓帆 1韦少钗1
作者信息
- 1. 广东潮州卫生健康职业学院,广东 潮州 521000
- 折叠
摘要
为缓解YOLOv7 在检测个人防护用品时面临标签重写、标签分配不平衡和特征耦合等问题,提出一种基于改进YOLOv7 的检测方法.首先去除YOLOv7 的大尺度和中尺度输出层,以降低标签重写率,且保证输出层得到充分训练;其次将输出层的定位和分类解耦,避免不同任务的特征表示互相影响,并选择在边界框级别检测防护服,在关键点级别检测防护帽和防护手套;最后引入部分卷积,实现实时检测.为验证该方法的有效性,使用实验人员穿戴防护用品的图像数据对所提方法进行验证.结果表明,相比YOLOv7,该方法的精确率和召回率分别提高了 4.1 和 4.5 个百分点,FPS(Frames Per Second)提升了 1.3 帧,可满足实验室场景下的个人防护用品穿戴检测需求.
Abstract
To alleviate the problems of label rewriting,unbalanced label assignment and feature coupling faced by YOLOv7 in detecting personal protective equipment,an improved YOLOv7 detection method is proposed.Firstly,the large-scale and medium-scale output layers of YOLOv7 are removed to reduce the label rewrite rate and to ensure that the output layer is adequately trained;secondly,the localization and classification of the output layer are decoupled to avoid that the feature representations of the different tasks affect each other and choose to detect the protective clothing at the bounding-box level,and the protective cap and protective gloves at the key-point level;Finally,a partial convolution is introduced to achieve real-time detection.In order to verify the effectiveness of the method,the proposed method is validated using image data of experimenters wearing protective equipment.The results show that compared with YOLOv7,the method improves the precision and recall by 4.1 and 4.5 percentage points,respectively,and the FPS is improved by 1.3 frames,which can satisfy the needs of personal protective equipment wearing detection in laboratory scenarios.
关键词
个人防护用品/穿戴检测/YOLOv7/单尺度输出/特征解耦/部分卷积Key words
personal protective equipment/detection of wearing/YOLOv7/single-scale output/feature decoupling/partial convolution引用本文复制引用
出版年
2024
NETL
NSTL
万方数据