基于注意力机制的YOLOv5口罩佩戴检测算法

【摘要】 <正>针对目前YOLO网络在口罩正确佩戴中检测率偏低的问题，提出了一种基于协同注意力机制和Bi FPN特征融合的改进YOLOv5口罩佩戴检测算法(YOLOv5-BC)。为了提高原始图像特征点的利用，将Bi FPN中的特征融合方法引入到Neck模块；将CA注意力机制引入YOLOv5网络用于增强口罩佩戴检测中对面部关键信息的提取。实验结果表明，改进的YOLOv5-BC算法在自制数据集上的检测精度达到了92.9%，相比于原始算法提高了1.3%，检测速度达到了60FPS，相较于主流算法具有较好的检测精度与实时性。更多还原