Skip to content

Latest commit

 

History

History
120 lines (70 loc) · 4.52 KB

README.md

File metadata and controls

120 lines (70 loc) · 4.52 KB

Pytorch Implementation of A Pyramid Architecture of GANs

This is a pytorch implementation for reproducing PAGAN results in the paper Learning Face Age Progression: A Pyramid Architecture of GANs.

Please note that this is not the official code and The code may still have errors for the results did not reach the original results in the paper.😩

Requirements

  • Pytorch 1.0
  • Python 3.6
  • Visdom 0.1.8
  • Pillow 6.0

Dataset

  • CACD
  • FGnet

Please pay attention to splitting CACD_dataset to train_dataset & val_dataset. and after make_label.py , move dataset to the path likedata_train/young(or elder1,elder2,elder3,val,test).

Pretrained Models

You can download pretrained vgg-face models from (http://www.robots.ox.ac.uk/~albanie/pytorch-models.html) and refer to this paper (https://arxiv.org/ftp/arxiv/papers/1709/1709.01664.pdf) to train age estimation networks, then move the two models to ./model_vgg.

It will require about 1.1 GB of disk space.

Running Models

you can run the shell script train.sh and test.sh.

Please note that modifying the path in the CONFIG when different age cluster.

Results

Here are some visualization results. And age estimation & face verification results by using face++ API.

  • train:

    • age_cluster_1:
      • original:
      • generate:
    • age_cluster_2:
      • original:
      • generate:
    • age_cluster_3:
      • original:
      • generate:

  • val(CACD):

    • val_age: 14

    • val_age: 22

    • val_age: 30

    • age estimation & face verification results:

      age cluster1 age cluster2 age cluster3
      average estimate age 42.1 50.7 61.7
      age accuracy(if estimate age in the age cluster ) 33.1% 33.0% 90.2%
      average verification confidence(with age cluster0) 91.5 86.6 79.9
      verification rate(FAR = 1e-5) 99.8% 97.8% 84.0%

  • test(FGnet):

    • test_age: 14

    • test_age: 22

    • test_age: 30

    • age estimation & face verification results:

      age cluster1 age cluster2 age cluster3
      average estimate age 37.6 48.4 51.1
      age accuracy(if estimate age in the age cluster ) 44.3% 42.7% 56.9%
      average verification confidence(with age cluster0) 92.3 87.7 87.7
      verification rate(FAR = 1e-5) 99.7% 98.1% 97.2%