[发明专利]一种基于生成对抗网络的人脸图像样本扩充方法

权利要求书

1.一种基于生成对抗网络的人脸图像样本扩充方法，包含以下步骤：

步骤1：获取原始人脸训练样本集合；

步骤2：将所述人脸训练样本集合中的人脸图像送入到S-Coder GAN(Source CoderGAN)生成对抗网络中，训练该生成对抗网络生成新的含有面部遮挡元素的人脸图像样本；其中：所述生成对抗网络的结构由以下模块组成：

模块1：使用跳层连接(Skip Connection)的残差结构设计的生成器模块(Generator)；

模块2：在所述生成对抗网络中加入基于独热+编码(One-Hot Encoding)算法构成的S-coder可控编码模块(S-Coder Module)；

模块3：使用Sigmoid层设计判别网络真假判别子模块，使用Softmax层设计判别网络标签判别子模块，以此两个子模块构建判别器模块(Discriminator)；

步骤3：使用周期一致的训练流程，即S-Coder GAN生成对抗网络中两组生成器组合，将生成图像反向送入结构相同的另一生成器得到重构图像，提高训练稳定性；

步骤4：最后使用训练完成的S-Coder GAN生成对抗网络，生成多种人脸遮挡物的人脸图像样本，得到扩充后的数据集。

2.如权利要求1所述基于生成对抗网络的人脸图像样本扩充方法，其特征在于，所述S-Coder GAN生成对抗网络的输入中使用向量掩码对输入训练目标域图像进行编码，并将编码与原域图像信息进行contact操作为新的输入送入生成器。

3.如权利要求1所述基于生成对抗网络的人脸图像样本扩充方法，其特征在于，所述S-Coder GAN生成对抗网络将生成的人脸图像送入S-Coder GAN判别器，此时判别器前部共享参数，用残差网络提取特征；随后将特征送入子网络a与子网络b；其中a网络是一个含有Sigmoid层网络输出图像的真假信息，另一个网络b含有Softmax层网络输出图像类别信息。

4.如权利要求1所述基于生成对抗网络的人脸图像样本扩充方法，其特征在于，所述S-Coder GAN生成对抗网络训练时使用两组相同生成网络组成类似周期一致生成对抗网络结构进行协同训练，使用周期一致性损失(cycle consistency loss)对重构图像与原图像进行约束，并在训练时使用Wasserstein loss对生成器与判别器进行优化。