ISSN 1009-5624 CN 10-2021/TQ    主管:中国乐凯集团有限公司    主办:北京乐凯科技有限公司

中国知网全文收录期刊
万方数据库收录期刊
RCCSE中文学术期刊
维普资讯网/超星域出版 全文收录
中国核心期刊(遴选)数据库收录期刊
首页 > 刊期 > 2022 > 7期 > 信息:技术与应用
基于改进 YOLOv3 的口罩佩戴检测算法
徐泽昊,刘 川

【摘要】针对防控工作中行人佩戴口罩检测无法实现高精度问题,在 YOLOv3 算法的基础上,对其进行了相应改进,设计了一种改进的 YOLOv3 口罩佩戴检测算法。首先,将 spp(spatial pyramid pooling) 结构添加到原 YOLOv3 结构中,解决输入神经网络不同图像尺寸不统一问题,然后利用 K-Means 算法对所构造的口罩数据进行聚类,重新获取检测的先验框,并根据图像大小进行合适的分配提升模型的精度。最后,再将原始 IOU Loss 损失函数替换为 CIOU Loss,解决边框回归时难以确认梯度下降方向的问题,降低检测中精度损失。经验证集测试可得,改良后的算法相比于原始 Y O L O v3算法 mAP 提高了 1.8%FPS 提高了 1.3%

【关键字】目标检测;YOLOv3;特征金字塔;损失函数