[发明专利]年龄预测模型的训练方法、年龄预测方法及相关装置

说明书

本申请实施例涉及人脸图像识别技术领域，公开了一种年龄预测模型的训练方法、年龄预测方法及相关装置，该年龄预测模型包括人脸编码网络、人脸解码网络以及感知机年龄预测网络，通过人脸编码网络对与原始人脸图像的年龄相同的年龄风格图像进行处理，并将处理之后的特征图进行风格融合，进一步通过人脸解码网络对融合后的特征图进行解码，并且，还通过感知机年龄预测网络对原始人脸图像进行年龄预测，构建年龄风格损失函数对年龄预测模型进行训练，本申请能够解决人脸特征的唯一性导致模型泛化能力不足的问题，提高年龄预测的准确度。

技术领域

本申请实施例涉及人脸图像识别技术领域，尤其涉及一种年龄预测模型的训练方法、年龄预测方法及相关装置。

背景技术

人脸图像包含了多种人脸特征信息，如人脸脸型、人脸皮肤状态、人脸表情、人脸五官、人脸年龄等，其中，人脸年龄作为一项较重要的特征信息，在人脸图像识别领域得到了广泛的应用。例如，在移动设备上运行的一些客户端上，具有人脸年龄识别的功能，其中，客户端通过获取人脸图像，并基于获取到的人脸图像输出识别得到的人脸年龄，以反馈给用户。

目前，人脸年龄估计方法通常利用卷积神经网络进行，通过构建数据集；将数据集划分为训练集、验证集；在网络输入层对mini-batch进行成对构建，然后送入两个孪生网络中进行训练；对VGG-16网络进行构建；网络训练；使用softmax loss和revisedcontrastiveloss共同作为监督信号对网络进行调整；网络测评；最终估计的年龄为softmax层得到的最大概率对应类别。

但是，这种方式对于输入的图像是采用不同年龄值的成对输入，通过共享网络参数，提取出人脸特征，再通过度量学习，计算成对样本的特征差异，而由于人脸特征具有唯一性，受到相同年龄具有不同人脸特征，导致模型在实际应用中泛化性能较差。

发明内容

本申请实施例提供一种年龄预测模型的训练方法、年龄预测方法及相关装置，以解决人脸特征的唯一性导致模型泛化能力不足的问题，提高年龄预测的准确度。

第一方面，本申请实施例中提供一种年龄预测模型的训练方法，年龄预测模型包括人脸编码网络、人脸解码网络以及感知机年龄预测网络，方法包括：

获取人脸数据集，其中，人脸数据集包括多个不同年龄值的人脸图像；

基于人脸编码网络，对人脸数据集中的原始人脸图像和年龄风格图像进行处理，得到原始人脸图像对应的特征图和年龄风格图像对应的特征图，其中，原始人脸图像和年龄风格图像对应的年龄值相同；

融合原始人脸图像对应的特征图与年龄风格图像对应的特征图，得到融合后的特征图；

将融合后的特征图输入到人脸解码网络，得到解码后的人脸图像；

将原始人脸图像对应的特征图输入到感知机年龄预测网络，其中，感知机年龄预测网络用于预测原始人脸图像对应的年龄；

构建年龄风格损失函数，基于人脸数据集和年龄风格损失函数，对年龄预测模型进行训练，直至年龄风格损失函数收敛，其中，年龄风格损失函数包括人脸解码网络输出的解码后的人脸图像和原始人脸图像的内容损失，以及，人脸解码网络输出的解码后的人脸图像和原始人脸图像的风格损失。

在一些实施例中，融合原始人脸图像对应的特征图与年龄风格图像对应的特征图，得到融合后的特征图，包括：

将原始人脸图像按照每一像素点的像素值的大小进行排序，并确定原始人脸图像对应的第一排序索引，其中，第一排序索引用于确定原始人脸图像中的每一像素点在排序之前的原始位置；

将年龄风格图像按照每一像素点的像素值的大小进行排序，并确定年龄风格图像对应的第二排序索引，其中，第二排序索引用于确定年龄风格图像中的每一像素点在排序之前的原始位置；

根据第一排序索引，对年龄风格图像进行处理，得到处理后的特征图；