[发明专利]一种图像处理模型的训练方法、图像处理模型及计算机设备

说明书

本申请实施例涉及图像处理技术领域，公开了一种图像处理模型的训练方法、图像处理模型及计算机设备，在经历多轮训练后，编码网络能够学习多个人在多个年龄段的年龄特征，并将各年龄段对应的年龄特征以编码形式进行表征，即每个年龄段的编码由编码网络学习多个人的年龄特征得到；生成式对抗网络通过将同一个人在各年龄段下的编码与人脸图像进行融合，学习同一个人在不同年龄段的年龄特征差异，使得生成的预测图像符合个体特性。此外，损失函数表征第一编码和第二编码之间的编码损失、真实人脸图像和预测人脸图像之间的特征损失和对抗损失，其中，真实人脸图像和预测人脸图像之间的特征损失使得生成式对抗网络中的生成器能够控制脸部特征的还原度。

技术领域

本申请实施例涉及图像处理技术领域，尤其涉及一种图像处理模型的训练方法、图像处理模型及计算机设备。

背景技术

随着机器学习技术不断成熟,基于机器学习技术的服务的种类也越来越多。例如计算机设备通过机器学习技术,能够对第一人脸图像进行处理,得到第二人脸图像。其中,第一人脸图像中的人脸对应的第一年龄和第二人脸图像中的人脸对应的第二年龄不同,但对应相同身份。上述服务在多个场景中被广泛需求，例如，在公安刑事侦破时，根据走失儿童的现有照片，推测其几年后的样貌，以便寻找；或者，根据嫌疑人年轻时的旧照片预测其逃亡多年后的照片；再例如，在影视业制作的过程中，当演员扮演年老角色时，根据演员现在的样貌推测其衰老后的样貌，方便化妆师画出年老角色形象；再例如，在休闲娱乐时，人们想通过当前的图像溯源自己小时候的照片。

计算机设备通常根据输入的第一人脸图像,基于用户的人脸年龄变化操作,通过机器学习模型对该第一人脸图像进行处理,得到第二人脸图像。其中,该人脸年龄变化操作用于指示计算机设备使第二人脸图像中的人脸对应的第二年龄大于第一人脸图像中的人脸对应的第一年龄,或小于第一年龄。该机器学习模型是基于不同年龄段的不同人脸图像训练得到的。

在通过上述方法对第一人脸图像进行处理时,特征变化仅由年龄段决定，通常使得个体的特征变化相同，即对于具有相同年龄的用户A和用户B，两者的图像从第一年龄变化为第二年龄时，他们的特征变化是相同的，特征变化单一，然而，在实际生活中，每个人随着年龄变化的特征变化受个体因素影响，具有特异性。

发明内容

本申请实施例主要解决的技术问题是提供一种图像处理模型的训练方法、图像处理模型及计算机设备，该方法训练得到的图像处理模型能够使得基于年龄变化的特征变化符合个体特性，能够更加准确地预测衰老图像或溯源年轻图像。

为解决上述技术问题，第一方面，本申请实施例中提供一种图像处理模型的训练方法，图像处理模型包括编码网络和生成式对抗网络，该方法包括：

获取真实人脸图像、训练人脸图像和与训练人脸图像对应的预期年龄，训练人脸图像与真实人脸图像反映同一人的人脸，真实人脸图像标注有年龄段，预期年龄位于真实人脸图像标注的年龄段内，预期年龄与训练人脸图像对应的年龄不同；

采用编码网络对真实人脸图像进行特征编码，得到第一编码，第一编码反映真实人脸图像在预期年龄下的人脸特征；

采用生成式对抗网络对第一编码和训练人脸图像进行特征融合，得到预测人脸图像，预测人脸图像为训练人脸图像融合有第一编码的特征后生成的图像；

利用损失函数对图像处理模型进行迭代训练，返回获取真实人脸图像、训练人脸图像和与训练人脸图像对应的预期年龄的步骤，直至图像处理模型收敛，其中，损失函数用于表征第一编码和第二编码之间的编码损失、真实人脸图像和预测人脸图像之间的特征损失和对抗损失，第二编码为采用编码网络对预测人脸图像进行特征编码获得的编码，对抗损失为所述生成式对抗网络计算得到的损失。

在一些实施例中，生成式对抗网络包括生成器，生成器包括依次设置的多个下采样层、多个深度层和多个上采样层；