吴慧婕, 赵刚, 胡送惠, 王正水. 基于YOLOv5s + CANet的人脸口罩检测改进算法[J]. 南昌航空大学学报(自然科学版), 2022, 36(4): 108-115. DOI: 10.3969/j.issn.2096-8566.2022.04.015
引用本文: 吴慧婕, 赵刚, 胡送惠, 王正水. 基于YOLOv5s + CANet的人脸口罩检测改进算法[J]. 南昌航空大学学报(自然科学版), 2022, 36(4): 108-115. DOI: 10.3969/j.issn.2096-8566.2022.04.015
Hui-jie WU, Gang ZHAO, Song-hui HU, Zheng-shui WANG. Improved Face Mask Detection Algorithm Based on YOLOv5s+CANet[J]. Journal of nanchang hangkong university(Natural science edition), 2022, 36(4): 108-115. DOI: 10.3969/j.issn.2096-8566.2022.04.015
Citation: Hui-jie WU, Gang ZHAO, Song-hui HU, Zheng-shui WANG. Improved Face Mask Detection Algorithm Based on YOLOv5s+CANet[J]. Journal of nanchang hangkong university(Natural science edition), 2022, 36(4): 108-115. DOI: 10.3969/j.issn.2096-8566.2022.04.015

基于YOLOv5s + CANet的人脸口罩检测改进算法

Improved Face Mask Detection Algorithm Based on YOLOv5s+CANet

  • 摘要: 为解决实际应用场景中人脸检测可能会出现的光照模糊、遮挡物不明确等一系列问题,提高检测的精确度,本文基于YOLOv5s算法,针对人脸口罩目标检测进行网络模型改进:在YOLOv5s的主干网络部分和颈部的不同网络层级处加入注意力机制,针对边界框回归任务,替换YOLOv5s模型的损失函数,加速收敛提高回归精度。实验结果表明,当在YOLOv5s的backbone部分的P5层和Neck部分的P4、P5层之后加入Coordinate 注意力机制,并用SIoU_Loss替换原本损失函数后,改进后的YOLOv5s算法与基准模型相比,精度值提升了1.6%。

     

    Abstract: In order to address a series of problems such as blurred lighting and unclear occlusions that may occur in face detection in practical application scenarios, and promote the detection accuracy, the network model for face mask object detection was engineered based on the YOLOv5s algorithm, and through the following strategies: Adding an attention mechanism to the backbone network part of YOLOv5s and different network levels of the neck; Replacing the loss function of the YOLOv5s model for the bounding box regression task to accelerate convergence and improve regression accuracy. The experimental results show that when the Coordinate attention mechanism is added after the P5 layer of the backbone part in YOLOv5s and the P4 and P5 layers of the Neck part, and the original loss function is replaced with SIoU Loss, the accuracy rendered by the engineered YOLOv5s algorithm is increased by 1.5% compared with that of the benchmark model.

     

/

返回文章
返回