[发明专利]为人脸识别生成训练数据的方法、装置、设备和存储介质

权利要求书

1.一种为人脸识别生成训练数据的方法，包括：

获取人脸识别失败时的待测人脸图像；

如果所述待测人脸图像为戴口罩的待测人脸图像，则提取所述待测人脸图像中的口罩图像；

获取未戴口罩的人脸图像；

计算所述未戴口罩的人脸图像的空间偏移角度；

根据所述空间偏移角度对所述口罩图像进行旋转以使所述口罩图像与所述未戴口罩的人脸图像在空间上一致；以及

将旋转之后的口罩图像融合至所述未戴口罩的人脸图像，以生成戴口罩的人脸图像。

2.根据权利要求1所述的为人脸识别生成训练数据的方法，其中，所述计算所述未戴口罩的人脸图像的空间偏移角度，包括：

获取标准正脸图像；以及

根据所述未戴口罩的人脸图像和所述标准正脸图像生成所述空间偏移角度。

3.根据权利要求2所述的为人脸识别生成训练数据的方法，其中，所述根据所述未戴口罩的人脸图像和所述标准正脸图像生成所述空间偏移角度，包括：

获取所述未戴口罩的人脸图像中的多个第一关键点位置；

获取所述标准正脸图像中的多个第二关键点位置，其中，所述多个第一关键点位置与所述多个第二关键点位置一一对应；以及

根据所述第一关键点位置和所述第二关键点位置计算所述空间偏移角度。

4.根据权利要求1所述的为人脸识别生成训练数据的方法，其中，所述根据所述空间偏移角度对所述口罩图像进行旋转以使所述口罩图像与所述未戴口罩的人脸图像在空间上一致，包括：

获取所述口罩图像的多个第三关键点位置；

根据所述空间偏移角度确定所述多个第三关键点位置的位置向量；以及

根据所述位置向量控制所述多个第三关键点位置进行移动，以使所述口罩图像与所述未戴口罩的人脸图像在空间上一致。

5.根据权利要求1-4任一项所述的为人脸识别生成训练数据的方法，其中，所述提取所述待测人脸图像中的口罩图像，包括：

获取所述待测人脸图像之中口罩区域的标注边界坐标；以及

根据所述口罩区域的标注边界坐标提取所述口罩图像。

6.根据权利要求1所述的为人脸识别生成训练数据的方法，其中，在所述根据所述空间偏移角度对所述口罩图像进行旋转以使所述口罩图像与所述未戴口罩的人脸图像在空间上一致之后，还包括：

对所述旋转之后的口罩图像进行数据增广，以得到多个不同形态的口罩图像，其中，所述多个不同形态的口罩图像用于与所述未戴口罩的人脸图像进行融合。

7.根据权利要求6所述的为人脸识别生成训练数据的方法，其中，所述对所述旋转之后的口罩图像进行数据增广，以得到多个不同形态的口罩图像，包括：

提取所述旋转之后的口罩图像中的口罩特征信息；以及

根据所述口罩特征信息和预设选取规则，控制所述旋转之后的口罩图像分别进行平移、旋转、大小缩放和颜色替换中的一种或多种，以得到所述多个不同形态的口罩图像。

8.根据权利要求1所述的为人脸识别生成训练数据的方法，还包括：

将所述未戴口罩的人脸图像与所述戴口罩的人脸图像进行绑定，并输入至预设的人脸识别模型以对所述预设的人脸识别模型进行更新。

9.根据权利要求1-4任一项所述的为人脸识别生成训练数据的方法，其中，所述空间偏移角度包括俯仰角、偏航角和翻滚角。