摘要
呼吸道飞沫是新型冠状病毒传播的主要媒介之一,通过口罩佩戴检测可以提醒人们及时佩戴口罩防止交叉感染。针对人员密集场所存在遮挡、小尺度目标检测困难的问题,提出基于改进YOLOv4的口罩佩戴检测算法。在YOLOv4中引入注意力机制,增强模型对有用特征的利用能力;复用空间金字塔池化,提高模型在不同感受野提取并融合空间特征的能力。实验结果表明,改进后的YOLOv4算法在口罩佩戴检测任务中的平均准确率均值达到了94.81%,相较于原始YOLOv4算法提高了5.76%,检测速度可达39 FPS,能够满足实际应用需求。
- 单位