Skip to content

caffeinism/FaceGAN

Repository files navigation

Facial Conditional GAN with Cycle Consistency

Recently, a number of papers have attempted to transformation faces. However, those papers require a specific set of data, such as MultiPIE, which are heavy and expensive! I wanted to have a similar effect with plain data sets.

I am still working on it, and the method and code will commit later.

Approach

According to CycleGAN, cycle-consistency could be used to generate images of other domains that preserve features.

We want to convert faces within the same domain. The input is the face of the person and the output is the face of the person. Therefore, the generator will be a network using the face and facial landmark as arguments.

generator

According to cGAN, conditional images can be generated by providing condition to generator and discriminator.

loss_adv

Finally, we use a pretrained face recognition network to preserve features by using cycle-consistency between target features and input features.

loss_cycle

Requirements

  • Python3
  • Pytorch 1.0.0
  • TensorBoardX

Result (using VGGFace2)

viusalized

It is getting better.

Reference

LightCNN

CycleGAN

GAN_stability

TODO list

  • add pretrained weights
  • add video samples
  • add ambiguous conditional method

About

Facial Conditional GAN with Cycle Consistency

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages