摘要
本发明公开了一种基于平衡栈式生成式对抗网络的人脸属性编辑方法,包括步骤:1)获取包含人脸图像和属性标签的数据集,并进行预处理;2)根据人脸图像的尺寸,构建多个由成对生成器和判别器构成的条件生成式对抗网络;3)利用预处理的数据集,针对不同人脸属性,采用加权学习和残差图像生成方式,独立训练所有条件生成式对抗网络;4)将所有训练好的生成器堆叠起来形成栈式结构,针对预处理的未知人脸图像,依次编辑对应人脸属性。本发明在训练过程中运用加权学习和残差图像生成方式,在属性编辑过程中采用栈式结构,最终使得模型能够有效应对数据不平衡问题,增强对少数类样本的编辑能力,提升在属性无关区域的图象生成效果,避免了属性纠缠的问题。
- 单位