基于改进YOLOv5的轻量型口罩佩戴检测算法

作者:王圣雄; 刘瑞安*; 燕达; 黄玉兰; 胡昕
来源:计算机时代, 2023, (05): 109-112.
DOI:10.16644/j.cnki.cn33-1094/tp.2023.05.024

摘要

提出一种基于改进YOLOv5的轻量型口罩检测算法CG-YOLOv5s。结合卷积注意力机制和Ghost卷积等技术,在网络中加入CBAM-CSP、GRCM改进模块以增强特征提取能力,改善模型的计算性能,使用Alpha-CIoU损失并结合DIoU非极大抑制方法,进一步提升检测精度。结果表明,CG-YOLOv5s在对检测速度影响较小的情况下,获得了89.1%的检测精度,模型大小减少了19.63%,实现了模型轻量化的效果。