PFA-GAN

PFA-GAN copied to clipboard

PFA-GAN: Progressive Face Aging with Generative Adversarial Network

arxiv | doi

This paper has been accepted as a Regular Paper by IEEE Transactions on Information Forensics & Security.

Requirements

• PyTorch 1.51+ (Ubuntu, cuda 10.2)
• Nvidia Apex (install from source)

Training

1. Downloading the datasets.
2. Aligning and generating data using the ages estimated by Face++ APIs.
3. Training with four GPUs on single node:

python -m torch.distributed.launch --nproc_per_node=4 --master_port=17674 \\
      main.py --device_ids 0 1 2 3 --source 0

4. The pre-trained model is to come.

Citation

If you found this code or our work useful please cite us:

@article{huang2020pfa,
  title={{PFA-GAN}: Progressive Face Aging with Generative Adversarial Network},
  author={Huang, Zhizhong and Chen, Shouzhen and Zhang, Junping and Shan, Hongming},
  journal={IEEE Transactions on Information Forensics and Security},
  year={2021},
  publisher={IEEE}
}