[发明专利]生成对抗网络训练方法及装置、电子设备和存储介质

说明书

本公开涉及一种生成对抗网络训练方法及装置、电子设备和存储介质，通过确定真实人脸图像和第一风格图像，将真实人脸图像输入生成网络得到的第二风格图像。根据真实人脸图像的第一头发区域和第二风格图像的第二头发区域确定第一颜色差异，根据第一风格图像的第三头发区域和第二头发区域确定纹理差异，根据第一颜色差异和纹理差异调节生成对抗网络，得到用于生成风格图像的生成网络。

技术领域

本公开涉及计算机技术领域，尤其涉及一种生成对抗网络训练方法及装置、电子设备和存储介质。

背景技术

目前，人像风格转换由于风格多样，方便在云端移动端应用，有着巨大的应用场景和市场需求。人像风格转换主要内容是将用户的真人图像转换为相应风格的风格化图像。相关技术在进行人像风格转换时通常会在转换前后通过其他图像处理模块进行图像处理，增加处理耗时且鲁棒性较差。同时还会在转换模型中嵌入特定区域的转换模块进行约束，增加存储占用以及处理过程的耗时。

发明内容

本公开提出了一种生成对抗网络训练方法及装置、电子设备和存储介质。

根据本公开的第一方面，提供了一种生成对抗网络训练方法， 所述生成对抗网络包括生成网络和判别网络，所述方法包括：

获取真实人脸图像和第一风格图像，所述第一风格图像中包括风格人脸；

将所述真实人脸图像输入所述生成网络，得到第二风格图像；

根据所述真实人脸图像的第一头发区域和所述第二风格图像的第二头发区域，确定所述第二风格图像与所述真实人脸图像中头发的第一颜色差异；

根据所述第一风格图像的第三头发区域和所述第二风格图像的第二头发区域，确定所述第二风格图像与所述第一风格图像中头发的纹理差异；

根据所述第一颜色差异和所述纹理差异调节所述生成对抗网络，得到用于生成风格图像的生成网络。

在一种可能的实现方式中，所述根据所述真实人脸图像的第一头发区域和所述第二风格图像的第二头发区域，确定所述第二风格图像与所述真实人脸图像中头发的第一颜色差异，包括：

提取所述真实人脸图像中的第一头发区域，以及所述第二风格图像的第二头发区域；

分别提取所述第一头发区域和所述第二头发区域的浅层特征，得到第一浅层特征和第二浅层特征；

根据所述第一浅层特征和所述第二浅层特征的差异确定所述第一颜色差异。

在一种可能的实现方式中，所述根据所述第一风格图像的第三头发区域和所述第二风格图像的第二头发区域，确定所述第二风格图像与所述第一风格图像中头发的纹理差异，包括：

提取所述第一风格图像的第三头发区域，以及所述第二风格图像的第二头发区域；

分别提取所述第三头发区域和所述第二头发区域的深层特征，得到第一深层特征和第二深层特征；

根据所述第一深层特征和所述第二深层特征的差异确定所述纹理差异。

在一种可能的实现方式中，所述方法还包括：

根据所述判别网络确定所述第二风格图像与所述真实人脸图像中头发的第二颜色差异；

所述根据所述第一颜色差异和所述纹理差异调节所述生成对抗网络，得到用于生成风格图像的生成网络包括：

根据所述第一颜色差异、所述纹理差异和所述第二颜色差异调节所述生成对抗网络，得到用于生成风格图像的生成网络。

在一种可能的实现方式中，所述根据所述判别网络确定所述第二风格图像与所述真实人脸图像中头发的第二颜色差异，包括：

根据所述第一头发区域确定发色特征值；