[发明专利]红外口罩人脸图像的生成方法、装置和人脸识别系统

权利要求书

1.一种红外口罩人脸图像的生成方法，其特征在于，包括：

对可见光人脸图像的人脸添加口罩，得到可见光口罩人脸图像；

采用至少一张原始红外人脸图像和高清人脸数据集训练得到风格迁移网络，所述风格迁移网络用于将人脸图像转换为红外人脸图像，所述原始红外人脸图像为红外图像采集仪采集的人脸像；

将所述可见光口罩人脸图像输入所述风格迁移网络，生成红外口罩人脸图像。

2.根据权利要求1所述的方法，其特征在于，对可见光人脸图像的人脸添加口罩，得到可见光口罩人脸图像，包括：

将所述可见光人脸图像输入FMA-3D，生成所述可见光口罩人脸图像。

3.根据权利要求1所述的方法，其特征在于，采用至少一张原始红外人脸图像和高清人脸数据集训练得到风格迁移网络，包括：

获取隐式风格编码器和人脸图像生成器，所述隐式风格编码器用于将人脸图像编码成向量编码，所述人脸图像生成器用于将所述向量编码还原为所述人脸图像，所述隐式风格编码器和所述人脸图像生成器均为在所述高清人脸数据集上预训练得到的；

将所述原始红外人脸图像输入所述隐式风格编码器，生成第一向量编码；

将所述第一向量编码运用Style Mixing原理生成多组第二向量编码；

将多组所述第二向量编码输入所述人脸图像生成器，生成多个随机人脸图像；

将所述原始红外人脸图像和多个所述随机人脸图像输入判别网络，得到多个损失值，所述损失值与所述随机人脸图像一一对应；

在损失函数值大于对应阈值的情况下，调整所述人脸图像生成器的参数，直至所述损失函数值小于或者等于对应的所述阈值，得到红外人脸图像生成器，所述损失函数值为多个所述损失值的平均值；

所述隐式风格编码器和所述红外人脸图像生成器形成所述风格迁移网络，所述隐式风格编码器的输出为所述红外人脸图像生成器的输入。

4.根据权利要求3所述的方法，其特征在于，所述原始红外人脸图像有多个，所述第一向量编码与所述原始红外人脸图像一一对应。

5.根据权利要求1至4任一项所述的方法，其特征在于，在将所述可见光口罩人脸图像输入所述风格迁移网络，生成红外口罩人脸图像之后，所述方法还包括：

采用人脸识别数据集训练人脸识别模型，得到更新后的人脸识别模型，所述人脸识别数据集包括所述可见光人脸图像对应的图像数据、所述可见光口罩人脸图像对应的图像数据和所述红外口罩人脸图像对应的图像数据。

6.根据权利要求5所述的方法，其特征在于，所述见光口罩人脸图像对应的图像数据在所述人脸识别数据集中的比例为20％～30％，所述红外口罩人脸图像对应的图像数据在所述人脸识别数据集中的比例为20％～30％。

7.一种红外口罩人脸图像的生成装置，其特征在于，包括：

处理单元，用于对可见光人脸图像的人脸添加口罩，得到可见光口罩人脸图像；

第一训练单元，用于采用至少一张原始红外人脸图像和高清人脸数据集训练得到风格迁移网络，所述风格迁移网络用于将人脸图像转换为红外人脸图像，所述原始红外人脸图像为图像采集得到；

生成单元，用于将所述可见光口罩人脸图像输入所述风格迁移网络，生成红外口罩人脸图像。

8.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质包括存储的程序，其中，所述程序执行权利要求1至6中任意一项所述的方法。

9.一种处理器，其特征在于，所述处理器用于运行程序，其中，所述程序运行时执行权利要求1至6中任意一项所述的方法。

10.一种人脸识别系统，其特征在于，包括：人脸识别模型、一个或多个处理器，存储器，以及一个或多个程序，其中，所述一个或多个程序被存储在所述存储器中，并且被配置为由所述一个或多个处理器执行，所述一个或多个程序包括用于执行权利要求1至6中任意一项所述的方法。