基于YOLOv3的轻量化口罩佩戴检测算法

• 作者: 薄景文 张春堂
• 来源期刊: 电子测量技术
• 学科: 工学
• 关键词: 口罩检测 轻量化 YOLOv3 注意力机制 损失函数
• 年，卷（期）: 2021,（23）
• 所属期刊栏目: 信息技术及图像处理|Information Technology and Image Processing
• 页码范围: 105-110
• 页数: 6页
• 分类号: TP391.4
• 语种: 中文
• DOI: 10.19651/j.cnki.emt.2107568

摘要

当前疫情防控形势严峻,在人群密集场所进行实时快速的口罩佩戴检测可以有效降低病毒传播的风险.针对目前人工检测效率低的问题,提出一种基于YOLOv3的轻量化口罩佩戴检测算法.使用ShuffleNetv2替换原来的主干特征提取网络,降低网络参数量,减少计算功耗.提出将SKNet注意力机制引入到特征融合网络部分,增强不同尺度的特征提取能力;使用CIoU作为边界框回归损失函数,进一步提高检测精度.在构建的人脸口罩检测数据集上实验表明,与原YOLOv3相比,所提算法在保持较高检测精度的情况下,检测速度提高了 34 FPS,有效地实现了准确快速的口罩佩戴检测,与其他主流目标检测算法相比,该算法也具有更好的检测效果.