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

权利要求书

1.一种生成对抗网络训练方法，其特征在于，所述生成对抗网络包括生成网络和判别网络，所述方法包括：

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

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

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

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

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

2.根据权利要求1所述的方法，其特征在于，所述根据所述真实人脸图像的第一头发区域和所述第二风格图像的第二头发区域，确定所述第二风格图像与所述真实人脸图像中头发的第一颜色差异，包括：

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

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

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

3.根据权利要求1所述的方法，其特征在于，所述根据所述第一风格图像的第三头发区域和所述第二风格图像的第二头发区域，确定所述第二风格图像与所述第一风格图像中头发的纹理差异，包括：

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

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

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

4.根据权利要求1所述的方法，其特征在于，所述方法还包括：

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

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

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

5.根据权利要求4所述的方法，其特征在于，所述根据所述判别网络确定所述第二风格图像与所述真实人脸图像中头发的第二颜色差异，包括：

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

将所述发色特征值和所述第二风格图像输入判别网络，得到所述第二风格图像与所述真实人脸图像中头发的第二颜色差异。

6.根据权利要求5所述的方法，其特征在于，所述根据所述第一头发区域确定发色特征值包括：

根据所述第一头发区域中各像素颜色确定发色特征值。

7.根据权利要求4-6中任意一项所述的方法，其特征在于，所述方法还包括：

根据所述第一风格图像和所述第二风格图像确定图像损失；

将所述第一风格图像和所述第二风格图像输入判别网络，得到对抗损失；

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

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

8.一种风格图像生成方法，其特征在于，所述方法包括：

获取待转化人脸图像；

将所述待转化人脸图像输入生成网络中，输出风格人脸图像，所述生成网络通过权利要求1-7中任意一项所述的生成对抗网络训练方法训练得到。