[发明专利]人脸特征提取模型的训练方法和人脸特征提取方法在审
申请号: | 202111193958.3 | 申请日: | 2021-10-13 |
公开(公告)号: | CN113947140A | 公开(公告)日: | 2022-01-18 |
发明(设计)人: | 彭楠;李弼;希滕;张刚 | 申请(专利权)人: | 北京百度网讯科技有限公司 |
主分类号: | G06K9/62 | 分类号: | G06K9/62;G06V10/764;G06V10/774;G06V40/16 |
代理公司: | 北京清亦华知识产权代理事务所(普通合伙) 11201 | 代理人: | 罗岚 |
地址: | 100085 北京市*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 特征 提取 模型 训练 方法 | ||
1.一种人脸特征提取模型的训练方法,包括:
获取包括人脸的样本图像;
将所述样本图像输入至人脸特征提取模型中,并由所述人脸特征提取模型中的主干网络提取所述样本图像的人脸特征;
获取所述人脸特征对应的协方差矩阵,并根据所述协方差矩阵获取第一损失函数;
根据所述人脸特征进行分类识别,并根据分类识别结果获取第二损失函数;
根据所述第一损失函数和所述第二损失函数调整所述人脸特征提取模型的模型参数,直至满足训练结束条件得到目标人脸特征提取模型。
2.根据权利要求1所述的训练方法,其中,所述根据所述人脸特征,获取所述人脸特征对应的协方差矩阵,并根据所述协方差矩阵获取第一损失函数,包括:
获取所述人脸特征的任一特征维度与其他特征维度之间协方差;
根据所有的所述协方差,生成所述协方差矩阵以及所述协方差矩阵对应的单位矩阵;
根据所述协方差矩阵和所述单位矩阵,获取所述第一损失函数。
3.根据权利要求2所述的训练方法,其中,所述根据所述协方差矩阵和所述单位矩阵,获取所述第一损失函数,包括:
获取所述协方差矩阵与所述单位矩阵之间的差值,并获取所述差值的平方值作为所述第一损失函数。
4.根据权利要求1所述的训练方法,其中,所述根据所述人脸特征进行分类识别,并根据分类识别结果获取第二损失函数,包括:
将所述人脸特征输入至所述人脸特征提取模型中的分类网络,由所述分类网络对所述人脸特征进行分类识别,以获取所述人脸特征的所述分类识别结果;
根据所述分类识别结果和所述样本图像的人脸特征的标记类别,获取所述第二损失函数。
5.根据权利要求4所述的训练方法,其中,所述根据所述人脸特征类别,获取所述第二损失函数,包括:
获取所述人脸特征类别与每个候选人脸特征类别之间的相似度;
根据所述相似度,获取所述人脸特征的所述分类识别结果;
根据所述标记类别和所述分类识别结果,获取交叉熵损失函数作为所述第二损失函数。
6.根据权利要求1-5中任一项所述的训练方法,其中,所述根据所述第一损失函数和所述第二损失函数调整所述人脸特征提取模型的模型参数,包括:
获取所述第一损失函数和所述第二损失函数之和作为目标损失函数;
根据所述目标损失函数,调整所述人脸特征提取模型的所述模型参数。
7.一种人脸特征提取方法,包括:
获取待提取图像;
将所述待提取图像输入至目标人脸特征提取模型,由所述目标人脸特征提取模型输出所述待提取图像的目标人脸特征,其中,所述目标人脸特征提取模型为采用如权利要求1-6任一项所述的训练方法训练出的模型。
8.根据权利要求7所述的提取方法,其中,所述由所述目标人脸特征提取模型输出所述待提取图像的目标人脸特征,包括:
由所述目标人脸特征提取模型中的主干网络提取所述待提取图像的所述目标人脸特征。
9.根据权利要求7或8所述的提取方法,其中,所述获取到所述目标人脸特征之后,还包括:
获取至少一个候选图像以及每个所述候选图像的候选人脸特征;
获取所述目标人脸特征与所有的所述候选人脸特征之间的相似度;
响应于所述相似度达到预设相似度阈值,则确定所述待识别图像对应的人脸与所述候选图像对应的人脸一致。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京百度网讯科技有限公司,未经北京百度网讯科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202111193958.3/1.html,转载请声明来源钻瓜专利网。