[发明专利]特征提取模型训练方法、人脸识别方法、装置、设备及介质

说明书

本发明提供一种特征提取模型训练方法、人脸识别方法、装置、设备及介质，该训练方法包括：获取样本数据集，所述样本数据集包含标注有对应身份标签的若干人脸图像，所述若干人脸图像分为被遮挡物部分遮挡的人脸图像、以及未遮挡的人脸图像；对所述样本数据集进行数据增强处理；对数据增强处理后的所述样本数据集进行图像截取处理，以根据预设截取规则，随机截取所述样本数据集中各人脸图像的局部区域；根据图像截取处理后的样本数据集对预先建立的特征提取模型进行训练，得到目标特征提取模型。本发明可提高部分遮挡人脸识别的准确性。

技术领域

本发明涉及人脸识别领域，尤其涉及一种特征提取模型训练方法、人脸识别方法、装置、设备及介质。

背景技术

受新冠肺炎的影响，目前人们无论进出公司还是小区通常都佩戴有口罩，这给基于人脸识别的门禁系统等带来了极大的挑战。因为口罩遮挡了部分人脸，而遮挡的人脸区域很难提取出真实准确的特征描述，而特征描述的不准确性大幅度增加了人脸识别技术的难度。对此情况，现有人脸识别方法无法满足准确识别要求。

发明内容

针对上述现有技术的不足，本发明的目的在于提供一种特征提取模型训练方法、人脸识别方法、装置、设备及介质，以提高部分遮挡人脸识别的准确性。

为了实现上述目的，本发明提供一种特征提取模型训练方法，包括：

获取样本数据集，所述样本数据集包含标注有对应身份标签的若干人脸图像，所述若干人脸图像分为被遮挡物部分遮挡的人脸图像、以及未遮挡的人脸图像；

对所述样本数据集进行数据增强处理；

对数据增强处理后的所述样本数据集进行图像截取处理，以根据预设截取规则，随机截取所述样本数据集中各人脸图像的局部区域；

根据图像截取处理后的样本数据集对预先建立的特征提取模型进行训练，得到目标特征提取模型。

在本发明一个优选实施例中，所述对所述样本数据集进行数据增强处理的步骤包括采用以下方式中的任何一种或多种得到新的人脸图像：

改变所述人脸图像的属性参数；

将所述人脸图像进行翻转；

从部分遮挡的人脸图像中提取遮挡物，并将提取到的遮挡物进行仿射变换后覆盖至未遮挡的人脸图像的对应位置；

学习所述人脸图像的特征，并利用部分遮挡的人脸图像中遮挡区域的特征替换未遮挡的人脸图像中对应区域的特征；

当某部分遮挡的人脸图像和某未遮挡的人脸图像标注有相同的身份标签时，从所述某未遮挡的人脸图像中截取与所述某部分遮挡的人脸图像中遮挡位置对应的图像，并将截取到的图像覆盖至所述某部分遮挡的人脸图像中的遮挡物上。

在本发明一个优选实施例中，所述根据图像截取处理后的样本数据集对预先建立的特征提取模型进行训练的步骤包括：

将所述人脸图像的局部区域输入所述特征提取模型进行处理，得到所述人脸图像的局部特征；

将所述人脸图像的局部特征输入预先训练的分类器，得到所述人脸图像的身份识别结果；

基于所述人脸图像对应的身份识别结果和身份标签，获取第一损失函数；

根据所述第一损失函数对所述特征提取模型进行迭代训练。

在本发明一个优选实施例中，在得到所述人脸图像的局部特征后，所述方法还包括：

将所述人脸图像的局部特征输入预设的二分类网络，以得到所述人脸图像是否被遮挡的遮挡判定结果；

基于所述人脸图像的遮挡判定结果和实际遮挡情况，获取第二损失函数；