• 融合注意力机制和快速网络的口罩人脸检测算法

    Mask face detection algorithm integrating fast network and attention mechanism

    • 口罩人脸检测是智能监控系统中的关键部分,在城市管理和公共卫生安全等方面有着重要意义。针对口罩人脸检测在处理光照条件、遮挡等问题的图像时出现的漏检和检测不准确问题,提出了FSR-YOLOv8口罩人脸检测算法。该算法通过在骨干网络中融合了优化的FasterNeXt,从而提高了检测速度和减少了参数量。同时,通过在特征融合点间加入改进的SPCBAM注意力机制,模型能更有效地提取关键特征,减少背景干扰。此外,构建了RFB_HDC_BD模块显著提高了特征语义信息的利用率。在同等情况下,FSR-YOLOv8优于YOLOv8以及其他主流算法。该模型在公开的 AIZOO 数据集上的 mAP值达到了 96.2%,在自制数据集上的mAP值达到了 89.2%,且模型参数量相比于YOLOv8模型降低了16%,参数量减低的同时具有较高的精确度。

       

      Abstract: Face mask detection is a crucial component of intelligent surveillance systems and is of great significance in urban management and public health security. Aiming at the problems of missed detection and inaccurate detection that occur when dealing with images with issues such as lighting conditions and occlusion in mask face detection, the FSR-YOLOv8 face mask detection algorithm is proposed. This algorithm integrates an optimized version of the FasterNext backbone network to improve detection speed and reduce parameter quantity. Additionally, an improved SPCBAM (Spatial Pyramid Contextual Attention Module) attention mechanism is added between feature fusion points, allowing the model to more efficiently extract key features and minimize background interference. Furthermore, the construction of the RFB_HDC_BD (Receptive Field Block with Horizontal Dense Connections and Block Decimation) module significantly increases the utilization rate of feature semantic information. Under the same conditions, FSR-YOLOv8 outperforms YOLOv8 and other mainstream algorithms. Moreover, the model achieves a mean Average Precision (mAP) value of 96.2% on the publicly available AIZOO dataset and an mAP of 89.2% on a custom dataset, while the model parameter count is reduced by 16% compared to the YOLOv8 model. While the number of parameters was reduced, it had a relatively high accuracy.

       

    /

    返回文章
    返回