Official implementation for paper High Resolution Face Age Editing
Official implementation for paper High Resolution Face Age Editing.
You can also create a new environment for this repo by running
conda env create -f env.yml
You can download the pretrained model by running:
cd ./logs/001 ./download.sh
Upload test images in the folder
/test/inputand run the test file. The output images will be saved in the folder
/test/output. You can change the desired target age with
--target_age.
python test.py --config 001 --target_age 65
Pretrained age classifier
To get age information, we use an age classifier pretrained on IMDB-WIKI dataset. We use the model released from paper Deep expectation of real and apparent age from a single image without facial landmarks by Rothe et al.
To prepare the model, you need to download the original caffe model and convert it to PyTorch format. We use the converter caffemodel2pytorch released by Vadim Kantorov. Then name the PyTorch model as
dex_imdb_wiki.caffemodel.ptand put it in the folder
/models.
Preparing your dataset
Download FFHQ dataset and unzip it to the
/data/ffhqdirectory.
Download age label to the
/datadirectory.
You can also train the model with your own dataset. Put your images in the
/datadirectory. With the pretrained classifier, you can create a new label file with the age of each image.
Training
You can modify the training options of the config file in
configsdirectory.
python train.py --config 001
We also provide a colab version for quick test. To run it using Google Colab, please click here.
@article{yao2020high, title = {High Resolution Face Age Editing}, author = {Xu Yao and Gilles Puy and Alasdair Newson and Yann Gousseau and Pierre Hellier}, journal = {CoRR}, volume = {abs/2005.04410}, year = {2020}, }
Copyright © 2020, InterDigital R&D France. All rights reserved.
This source code is made available under the license found in the LICENSE.txt in the root directory of this source tree.