多尺度通道注意力机制的口罩佩戴检测算法

作者：李莉; 刘阳; 王巍; 耿华; 李丽宏

来源：计算机工程与设计, 2023, 44(02): 598-604.

DOI：10.16208/j.issn1000-7024.2023.02.038

• 口罩佩戴检测
• 特征提取
• YOLOv3
• 注意力机制
• 聚类
• 损失函数
• 卷积神经网络

为解决密集人群场景下口罩佩戴检测出现的关联特征信息利用率不足导致边界框定位不精准的问题，提出一种基于多尺度通道注意力机制的YOLO-Mask改进算法。利用通道注意力机制重构YOLOv3特征提取网络，促使网络对关联目标区域具有更高的关注度；采用K-means++聚类算法对数据集进行聚类优化，提高检测准确率；对损失函数进行优化。在PASCAL VOC和RFMD(real-world masked face dataset)数据集上的实验结果表明，YOLO-Mask算法与其它算法相比，具有更好的检测效果，平均精度分别为81.3%、85.3%,相比原始YOLOv3算法分别提高了3.0%和3.8%。

• 单位
  河北工程大学