[发明专利]人脸旋转模型的生成方法、装置、计算机设备及存储介质

说明书

本发明公开了一种人脸旋转模型的生成方法、装置、计算机设备及存储介质。所述方法包括：获取人脸训练图片；根据三维人脸参数以及预设的平均人脸模型获取真实位置图；提取人脸训练图片中的目标特征，将目标特征输入至人脸旋转模型中，并获取人脸旋转模型输出的所有第二关键点的三维位置信息；通过掩码设置各第一关键点的三维位置信息以及各第二关键点的三维位置信息在均方误差损失函数中所占的权重值，并将各第一关键点的三维位置信息以及各第二关键点的三维位置信息输入至均方误差损失函数中，得到一个损失结果；在损失结果小于或等于预设损失值时，确认人脸旋转模型训练完成，通过本发明可精准计算人脸照片旋转为正脸照片所需的旋转角度。

技术领域

本发明涉及模型构建领域，尤其涉及一种人脸旋转模型的生成方法、装置、计算机设备及存储介质。

背景技术

对于非正面人脸来说，人工智能领域的人脸识别技术还不够成熟，非正面人脸的人脸识别技术也一直是人脸识别领域的难题。在现有人脸识别系统中的非正面人脸的人脸对齐技术，通常只能解决人脸姿态中的roll角的旋正，而对于大角度的yaw角度或者pitch角度直接使用相似变换，如此，对齐后人脸将会产生较大的形变，不利于后续的人脸识别；同时，现有技术中的非正面人脸的人脸对齐也无法获取到人脸的准确三维坐标信息。因此，寻找一种可解决上述问题的技术方案成为本领域技术人员亟解决的问题。

发明内容

基于此，有必要针对上述技术问题，提供一种人脸旋转模型的生成方法、装置、计算机设备及存储介质，解决目前不能较精准计算人脸照片旋转为正脸照片所需的旋转角度的问题。

一种人脸旋转模型的生成方法，包括：

从目标集合中获取人脸训练图片；所述人脸训练图片中已标注三维人脸参数；

获取预设的平均人脸模型，根据所述三维人脸参数以及所述预设的平均人脸模型获取真实位置图；所述真实位置图中包含与所述人脸训练图片中的人脸图像对应的正面人脸以及所述正面人脸的所有第一关键点的三维位置信息；

提取所述人脸训练图片中的所述人脸图像的目标特征，将所述目标特征输入至包含初始参数的人脸旋转模型中，并获取所述人脸旋转模型输出的所有第二关键点的三维位置信息；

通过掩码设置各所述第一关键点的三维位置信息以及各所述第二关键点的三维位置信息在均方误差损失函数中所占的权重值，并将各所述第一关键点的三维位置信息以及各所述第二关键点的三维位置信息输入至所述均方误差损失函数中，得到一个损失结果；

在所述损失结果小于或等于预设损失值时，确认所述人脸旋转模型训练完成；所述人脸旋转模型用于在将人脸照片输入该人脸旋转模型时，确认将该人脸照片旋转为正脸照片所需的旋转角度。

一种人脸旋转模型的生成装置，包括：

第一获取模块，用于从目标集合中获取人脸训练图片；所述人脸训练图片中已标注三维人脸参数；

第二获取模块，用于获取预设的平均人脸模型，根据所述三维人脸参数以及所述预设的平均人脸模型获取真实位置图；所述真实位置图中包含与所述人脸训练图片中的人脸图像对应的正面人脸以及所述正面人脸的所有第一关键点的三维位置信息；

第三获取模块，用于提取所述人脸训练图片中的所述人脸图像的目标特征，将所述目标特征输入至包含初始参数的人脸旋转模型中，并获取所述人脸旋转模型输出的所有第二关键点的三维位置信息；

输入模块，用于通过掩码设置各所述第一关键点的三维位置信息以及各所述第二关键点的三维位置信息在均方误差损失函数中所占的权重值，并将各所述第一关键点的三维位置信息以及各所述第二关键点的三维位置信息输入至所述均方误差损失函数中，得到一个损失结果；