Official implementation of Controllable 3D Generative Adversarial Face Model via Disentangling Shape and Appearance

Fariborz Teherkhani, Aashish Rai*, Shaunak Srivastava*, Quankai Gao*, Xuanbai Chen, Fernando de la Torre, Steven Song, Aayush Prakash, Daeil Kim

(* equal contribution)

More details coming soon!

[Project Page] [Video] [Colab Demo] [Demo Code] [Arxiv]

Install packages

See install.md

Testing

Download pre-trained weights and put the "checkpoints" folder in the main directory. [Link]

• Generate 3D Faces (mesh and texture)

  python generate_faces.py
  

• Generate meshes only

  python test_gan3d.py
  

• Generate textures only

  python test_texture.py
  

Train your own model

Preprocess data

python preprocess.py (Coming Soon)

Start training

• Shape

  Train AE
  python train_ae.py 
  

  Train GAN
  python train_gan3d.py 
  

• Texture

  Train P-GAN
  python train_texture.py --init_step 1 --batch_size 128