结合 ArcFace 与知识蒸馏的口罩人脸识别方法

MASK FACE RECOGNITION METHOD COMBINING ARCFACE AND KNOWLEDGE DISTILLATION

  • 摘要: 近几年由于疫情影响,人们在公共场所需严格佩戴口罩,而传统的人脸识别系统无法识别口罩人脸。针对该问题,在 ArcFace 的基础上做出改进,在人脸特征提取网络 IResNet 中级联一个眉眼注意力模块和两个 CBAM 模块,在该网络的基础上使用知识蒸馏的方法。既加快了模型的推理速度,又优化了网络的分类决策与特征映射层,使得戴或不戴口罩都保持同一身份的相似性。在六个不同的基准数据集上进行实验验证,结果表明,口罩人脸识别的精度与速度都有了较大的提升,增强了人脸识别模型在口罩人脸上的性能。

     

    Abstract: In recent years, due to the impact of the epidemic, people need to strictly wear masks in public places, while the traditional face recognition system can not recognize the face of masks. To solve this problem, this paper makes an improvement on the basis of ArcFace. An eyebrow eye attention module and two CBAM modules were linked in IResNet, the face feature extraction network. The method of knowledge distillation was used on the basis of this network. It not only accelerated the reasoning speed of the model, but also optimized the classification decision-making and feature mapping layer of the network, so that wearing or not wearing masks can maintain the same identity similarity. The experimental results on six different benchmark data sets show that the accuracy and speed of mask face recognition have been greatly improved, and the performance of face recognition model on mask face has been improved.

     

/

返回文章
返回