Skip to content

lumosity4tpj/Pytorch-Implementation-of-A-Pyramid-Architecture-of-GANs

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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 (https://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%

About

No description or website provided.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published