aminesoulaymani / cddfm3d

Cross-Domain and Disentangled Face Manipulation with 3D Guidance

Cross-Domain and Disentangled Face Manipulation with 3D Guidance

Project Page | Paper (ArXiv)

Can Wang¹, Menglei Chai², Mingming He³, Dongdong Chen⁴, Jing Liao^1
1City University of Hong Kong, ²Creative Vision, Snap Inc., ³USC Institute for Creative Technologies, ⁴Microsoft Cloud AI

Abstract

Face image manipulation via three-dimensional guidance has been widely applied in various interactive scenarios due to its semantically-meaningful understanding and user-friendly controllability. However, existing 3D-morphable-model-based manipulation methods are not directly applicable to out-of-domain faces, such as non-photorealistic paintings, cartoon portraits, or even animals, mainly due to the formidable difficulties in building the model for each specific face domain. To overcome this challenge, we propose, as far as we know, the first method to manipulate faces in arbitrary domains using human 3DMM. This is achieved through two major steps: 1) disentangled mapping from 3DMM parameters to the latent space embedding of a pre-trained StyleGAN2 that guarantees disentangled and precise controls for each semantic attribute; and 2) cross-domain adaptation that bridges domain discrepancies and makes human 3DMM applicable to out-of-domain faces by enforcing a consistent latent space embedding. Experiments and comparisons demonstrate the superiority of our high-quality semantic manipulation method on a variety of face domains with all major 3D facial attributes controllable – pose, expression, shape, albedo, and illumination. Moreover, we develop an intuitive editing interface to support user-friendly control and instant feedback.

Getting Started

0. Requirements

• Python 3.8
• Torch 1.7.1 or Torch 1.8.0
• Pytorch3D for rendering images

Note: Related checkpoints and 3DMM bases can be downloaded from here; you can also take the following steps to prepare these.

1. Data Preparation

1. Download the StyleGAN2 checkpoint from here and place it into the 'stylegan2-pytorch/checkpoint' directory.
2. For quickly trying our method, I recommend to generate 4K latent(StyleSpace)&image training pairs:

cd stylegan2-pytorch
python generate_data.py --pics 4000 --ckpt checkpoint/stylegan2-ffhq-config-f.pt

Once finished, you will acquire 'Images' and 'latents.pkl' files.

2. Estimate 3DMM Parameters and Facial Landmarks

1. Download checkpoint from here and place it into the 'Deep3DFaceReconstruction-pytorch/network' directory;
2. Download 3DMM bases from here and place these files into the 'Deep3DFaceReconstruction-pytorch/BFM' directory;
3. Estimate 3DMM parameters and facial Landmarks:

cd Deep3DFaceReconstruction
python extract_gt.py ../stylegan2-pytorch/Images

Once finished, you will acquire the 'params.pkl' file. Then estimate the landmarks using dlib and split training and testing datasets. Please download the dlib landmark predictor from here and place it into the 'sample_dataset' directory.

mv stylegan2-pytorch/Images sample_dataset/
mv stylegan2-pytorch/latents.pkl sample_dataset/
mv Deep3DFaceReconstruction-pytorch/params.pkl sample_dataset/
cd sample_dataset/
python extract_landmarks.py
python split_train_test.py

3. Training and Testing the Attribute Prediction Network

Copy BFM from Deep3DFaceReconstruction:

cp -r Deep3DFaceReconstruction-pytorch/BFM/ ./bfm/

Then train the Attribute Prediction Network:

python train.py --name apnet_wpdc --model APModel --train_wpdc

Test the Attribute Prediction Network and you will find results (the rendered image and the mesh) in the 'results' directory:

python evaluate.py --name apnet_wpdc --model APModel

Some testing results of APNet (pre-trained model with only 4K images, 3800 for training and 200 for testing):

Some important parameters for training or testing:

Parameter	Default	Description
--name	'test'	name of the experiment
--model	'APModel'	which model to be trained
--train_wpdc	False	whether use WPDC loss
--w_wpdc	1.0	weight of the WPDC loss
--data_dir	'sample_dataset'	path to dataset
--total_epoch	200	total epochs for training
--save_interval	20	interval to save the model
--batch_size	128	batch size to train the model
--load_epoch	-1 (the final saved model)	which checkpoint to load for testing

To Do

• Code for generating latent&image training pairs;
• Code for estimating 3DMM parameters and landmarks;
• Code and pre-trained models for the Attribute Prediction Network;
• Code and pre-trained models for the Latent Manipulation Network
• Code, Data and pre-trained models for Latent-Consistent Finetuning
• Remains ...

References

• StyleGAN2-pytorch: https://github.com/rosinality/stylegan2-pytorch
• Deep3DFaceReconstruction-pytorch: https://github.com/changhongjian/Deep3DFaceReconstruction-pytorch

Citation

If you find our work useful for your research, please consider citing the following papers :)

@article{wang2021cross,
  title={Cross-Domain and Disentangled Face Manipulation with 3D Guidance},
  author={Wang, Can and Chai, Menglei and He, Mingming and Chen, Dongdong and Liao, Jing},
  journal={arXiv preprint arXiv:2104.11228},
  year={2021}
}