[发明专利]一种实时更新余弦夹角损失函数参数的人脸识别方法

说明书

本发明提出了一种实时更新余弦夹角损失函数参数的人脸识别方法，属于计算机视觉中的人脸识别领域。该方法包括：（1）收集人脸图像，并将人脸图像按个体分类，并对每个人脸图像进行数据标注；（2）对人脸图像进行图像预处理，得到人脸图像数据集；（3）初始化余弦夹角损失函数的的余弦值放大尺度和余弦夹角间隔；（4）将图像数据集送入卷积神经网络，实时计算更新余弦值放大尺度和余弦夹角间隔，直至完成对卷积神经网络的训练，（5）将需要进行比对判断的人脸图像输入训练好的卷积神经网络中，输出人脸特征向量，用于进行人脸识别匹配。本发明的人脸识别方法具有训练收敛速度快，识别准确率高的特点。

技术领域

本发明属于计算机视觉中的人脸识别领域，尤其涉及到一种基于自适应学习机制实时更新余弦夹角损失函数参数的人脸识别方法。

背景技术

随着计算机视觉技术的发展以及日渐增长的智能安防、电子商务等需求，人脸识别技术成为人工智能应用产品落地最为广泛的一个领域。使用深度学习方法的人脸识别技术核心在于通过卷积神经网络（CNN）抽象化出人脸图像的特征，用于计算人脸图像之间的相似度，进而实现人脸识别的功能。

在人脸识别的实际应用中，往往受到摄像头成像、光照、面部遮挡等因素的影响，造成识别能力下降，从而影响最终识别结果。特别在金融、安防等重要场景，错误的人脸识别结果可能造成不可估量的影响。因而如何训练出一个能精确提取人脸特征的卷积神经网络成为计算机视觉技术中心一个重要的研究课题。在卷积神经网络训练过程中，损失函数的选取对最终的特征提取能力有决定性的影响。为了提升人脸识别模型的判别性能，近年来一系列加入间隔抑制的余弦夹角损失函数，如A-Softmax，CosFace, ArcFace开始成为训练人脸识别模型的主流方法。这些余弦夹角损失函数的共同之处在于均使用了余弦角间隔和对余弦值尺度放大的方式进行模型训练，然而模型的余弦角间隔值和余弦值放大的尺度值需要在训练前预先确定。这两个参数对模型训练的最终结果有决定性的影响，过小的参数值会造成模型的判别能力变弱，而过大的参数值虽然有利于提升模型的最终能力，但在训练初期会导致模型难以收敛。目前主流的做法是在人脸识别模型的训练任务中针对不同的参数选取进行多组对比性实验，依据实验结果选取合适的参数取值。然而这种方式随机性大，而且随着训练数据的迭代变换，之前获取的最佳取值可能会不适用于新的数据与网络。在网络训练的过程中，训练早期模型性能较弱，需要较小的参数值来帮助模型快速收敛，而在训练后期模型性能逐渐增强，这时需要更大的参数值来加强模型的判别能力。

为了解决上述问题，如何设计一种随着网络训练进程自动更新参数的余弦夹角损失函数成为进一步提升人脸识别模型性能的一个重要方向。

发明内容

本发明的目的在于针对现有技术的不足，提出了一种实时更新余弦夹角损失函数参数的人脸识别方法，用于人脸识别模型的训练。摆脱了传统余弦夹角损失函数需要预先确定间隔值和尺度值的限制，该人脸识别方法适用于不同的人脸识别网络和数据集，并且能随着训练进程自适应地调节参数值，加速卷积神经网络的收敛，该方法的人脸识别能力更强。

本发明为实现上述目的，采用了以下技术方案：一种实时更新余弦夹角损失函数参数的人脸识别方法，包括以下步骤：

（1）收集人脸图像，并将人脸图像按个体进行分类，并按照分类的结果对每个人脸图像进行数据标注；

（2）将步骤（1）收集的人脸图像进行水平翻转、去噪、亮度增强和对比度增强处理后，得到人脸图像数据集，将所述人脸图像数据集分批；

（3）初始化余弦夹角损失函数的余弦值放大尺度和余弦夹角间隔，根据初始化的余弦值放大尺度和余弦夹角间隔，得到余弦夹角损失函数；