一种跨域人脸识别方法技术

技术编号:39521956 阅读:23 留言:0更新日期:2023-11-25 19:01
本申请公开一种跨域人脸识别方法

【技术实现步骤摘要】
一种跨域人脸识别方法、装置、计算机设备及存储介质


[0001]本申请涉及人脸识别领域,具体涉及一种跨域人脸识别方法

装置

计算机设备及存储介质


技术介绍

[0002]随着人工智能的发展,人脸识别技术应用越来越广泛,人脸识别技术面临越来越多的跨域场景应用的问题

比较典型的是人证识别

特定人群的识别等应用场景(如小学生

老年人),这些特定域的数据量往往远小于通用人脸识别数据,导致直接基于这些数据训练人脸识别模型,容易出现过拟合的问题

同时,虽然这些特定域数据有自身特点,但与通用人脸数据也有比较大的联系,利用通用人脸数据来增强这些特定域的人脸识别性能是有意义的

[0003]针对这一问题,一种比较常用的方式是先使用通用人脸识别数据训练通用人脸识别模型,然后再到特定域数据上进行微调训练

但是这种方式存在特定域数据量较小,容易出现过拟合的问题,在学习过程中失去了通用人脸数据的相关知识

[0004]另一种方式是将通用人脸识别数据和特定域数据合并在一起进行训练,这种方式的缺点是:通用人脸识别数据量远多于特定域的数据,导致特定域的权重过低,虽然可以在计算损失函数时,人为调整不同数据的权重,但这种方式实现复杂,需要对数据做特殊标记和处理

[0005]因此,现有的人脸识别技术在跨域场景应用时存在准确率低的技术问题
/>[0006]本
技术介绍
描述的内容仅为了便于了解本领域的相关技术,不视作对现有技术的承认


技术实现思路

[0007]为了解决上述技术问题,本申请提供了一种跨域人脸识别方法

装置

计算机设备及存储介质,具体方案如下:第一方面,本申请实施例提供了一种跨域人脸识别方法,所述跨域人脸识别方法包括:预先构建初始双分支人脸识别模型并获取跨域人脸图像,其中,所述初始双分支人脸识别模型包括特征共享网络和双分支网络,所述双分支网络包括左分支网络和右分支网络,所述跨域人脸图像包括通用域人脸数据和特定域人脸数据;分别采用所述通用域人脸数据和所述特定域人脸数据训练所述初始双分支人脸识别模型;在所述初始双分支人脸识别模型中引入双重注意力模块,得到目标双分支人脸识别模型;基于所述通用域人脸数据和所述特定域人脸数据共同训练所述目标双分支人脸识别模型;
获取待识别人脸图像并将所述待识别人脸图像输入训练后的目标双分支人脸识别模型进行识别,得到识别结果,其中,所述待识别人脸图像包括所述通用域人脸数据和所述特定域人脸数据中的任一种

[0008]根据本申请公开的一种具体实施方式,分别采用所述通用域人脸数据和所述特定域人脸数据训练所述初始双分支人脸识别模型的步骤,包括:固定所述右分支网络,使用所述通用域人脸数据训练所述左分支网络和所述特征共享网络;固定所述左分支网络及所述特征共享网络,使用所述特定域人脸数据训练所述右分支网络

[0009]根据本申请公开的一种具体实施方式,所述初始双分支人脸识别模型为多阶段模块结构,分支节点前的各阶段模块构成所述特征共享网络,分支节点后的所述右分支网络和所述左分支网络均包含一个阶段模块

一个第一
Dropout
层和一个第一全连接层;在所述初始双分支人脸识别模型中引入双重注意力模块的步骤,包括:去除所述右分支网络和所述左分支网络中的所述第一
Dropout
层和所述第一全连接层;在所述右分支网络和所述左分支网络中的阶段模块后分别引入双重注意力模块,其中,所述双重注意力模块包括自注意力模块和通道注意力模块;将两个所述双重注意力模块的输出相加后,连接一个第二
Dropout
层和一个第二全连接层,其中,所述第二全连接层的权重初始化为两个所述第一全连接层的平均值

[0010]根据本申请公开的一种具体实施方式,基于所述通用域人脸数据和所述特定域人脸数据共同训练所述目标双分支人脸识别模型的步骤,包括:获取所述双分支网络中任一所述阶段模块输出的初始特征图,其中,所述跨域人脸图像由所述特征共享网络

所述双分支网络中的所述阶段模块进行特征提取和加深,得到所述初始特征图;将所述初始特征图输入所述自注意力模块,得到自注意力特征图;将所述自注意力特征图输入所述通道注意力模块,得到对应当前分支网络的加权特征图;将所述左分支网络的加权特征图和所述右分支网络的加权特征图相加后,依次输入所述第二
Dropout
层和所述第二全连接层,得到最终特征向量和训练后的所述目标双分支人脸识别模型

[0011]根据本申请公开的一种具体实施方式,所述自注意力模块包括不同类型的卷积,其中,所述不同类型的卷积包括查询卷积

键卷积

值卷积;将所述初始特征图输入所述自注意力模块,得到自注意力特征图的步骤,包括:所述初始特征图分别经由所述查询卷积

键卷积和值卷积变换为第一特征图

第二特征图和第三特征图;将所述第一特征图和第二特征图相乘并按行归一化得到注意力权重图;将所述第三特征图和所述注意力权重图加权相乘后,得到所述自注意力特征图

[0012]第二方面,本申请实施例提供了一种跨域人脸识别装置,所述跨域人脸识别装置包括:
获取模块,用于预先构建初始双分支人脸识别模型并获取跨域人脸图像,其中,所述初始双分支人脸识别模型包括特征共享网络和双分支网络,所述双分支网络包括左分支网络和右分支网络,所述跨域人脸图像包括通用域人脸数据和特定域人脸数据;单域训练模块,用于分别采用所述通用域人脸数据和所述特定域人脸数据训练所述初始双分支人脸识别模型;双重引入模块,用于在所述初始双分支人脸识别模型中引入双重注意力模块,得到目标双分支人脸识别模型;全域训练模块,用于基于所述通用域人脸数据和所述特定域人脸数据共同训练所述目标双分支人脸识别模型;识别模块,用于获取待识别人脸图像并将所述待识别人脸图像输入训练后的目标双分支人脸识别模型进行识别,得到识别结果,其中,所述待识别人脸图像包括所述通用域人脸数据和所述特定域人脸数据中的任一种

[0013]根据本申请公开的一种具体实施方式,所述单域训练模块具体用于:固定所述右分支网络,使用所述通用域人脸数据训练所述左分支网络和所述特征共享网络;固定所述左分支网络及所述特征共享网络,使用所述特定域人脸数据训练所述右分支网络

[0014]根据本申请公开的一种具体实施方式,所述初始双分支人脸识别模型为多阶段模块结构,分支节点前的各阶段模块构成所述特征共享网络,分支节点后的所述右分支网络和所述左分支网络均包含一个阶段模块

一个第一
Dropout
层和一个第一本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.
一种跨域人脸识别方法,其特征在于,所述跨域人脸识别方法包括:预先构建初始双分支人脸识别模型并获取跨域人脸图像,其中,所述初始双分支人脸识别模型包括特征共享网络和双分支网络,所述双分支网络包括左分支网络和右分支网络,所述跨域人脸图像包括通用域人脸数据和特定域人脸数据;分别采用所述通用域人脸数据和所述特定域人脸数据训练所述初始双分支人脸识别模型;在所述初始双分支人脸识别模型中引入双重注意力模块,得到目标双分支人脸识别模型;基于所述通用域人脸数据和所述特定域人脸数据共同训练所述目标双分支人脸识别模型;获取待识别人脸图像并将所述待识别人脸图像输入训练后的目标双分支人脸识别模型进行识别,得到识别结果,其中,所述待识别人脸图像包括所述通用域人脸数据和所述特定域人脸数据中的任一种;其中分别采用所述通用域人脸数据和所述特定域人脸数据训练所述初始双分支人脸识别模型的步骤,包括:固定所述右分支网络,使用所述通用域人脸数据训练所述左分支网络和所述特征共享网络;固定所述左分支网络及所述特征共享网络,使用所述特定域人脸数据训练所述右分支网络
。2.
根据权利要求1所述的跨域人脸识别方法,其特征在于,所述初始双分支人脸识别模型为多阶段模块结构,分支节点前的各阶段模块构成所述特征共享网络,分支节点后的所述右分支网络和所述左分支网络均包含一个阶段模块

一个第一
Dropout
层和一个第一全连接层;在所述初始双分支人脸识别模型中引入双重注意力模块的步骤,包括:去除所述右分支网络和所述左分支网络中的所述第一
Dropout
层和所述第一全连接层;在所述右分支网络和所述左分支网络中的阶段模块后分别引入双重注意力模块,其中,所述双重注意力模块包括自注意力模块和通道注意力模块;将两个所述双重注意力模块的输出相加后,连接一个第二
Dropout
层和一个第二全连接层,其中,所述第二全连接层的权重初始化为两个所述第一全连接层的平均值
。3.
根据权利要求2所述的跨域人脸识别方法,其特征在于,基于所述通用域人脸数据和所述特定域人脸数据共同训练所述目标双分支人脸识别模型的步骤,包括:获取所述双分支网络中任一所述阶段模块输出的初始特征图,其中,所述跨域人脸图像由所述特征共享网络

所述双分支网络中的所述阶段模块进行特征提取和加深,得到所述初始特征图;将所述初始特征图输入所述自注意力模块,得到自注意力特征图;将所述自注意力特征图输入所述通道注意力模块,得到对应当前分支网络的加权特征图;将所述左分支网络的加权特征图和所述右分支网络的加权特征图相加后,依次输入所
述第二
Dropout
层和所述第二全连接层,得到最终特征向量和训练后的所述目标双分支人脸识别模型
。4.
根据权利要求3所述的跨域人脸识别方法,其特征在于,所述自注意力模块包括不同类型的卷积,其中,所述不同类型的卷积包括查询...

【专利技术属性】
技术研发人员:张波刘浩徐宇杰叶剑赵拯
申请(专利权)人:杭州宇泛智能科技有限公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1