AACNN

Tensorflow Inplementation for AACNN : Attribute Augmented Convolutional Neural Network for Face Hallucination (NTIRE2018)

Paper

Attribute Augmented Convolutional Neural Network for Face Hallucination
Cheng-Han Lee ¹, Kaipeng Zhang ¹, Hu-Cheng Lee ¹, Chia-Wen Cheng ², and Winston H. Hsu ^1
1 National Taiwan University, ² The University of Texas at Austin
IEEE Conference on Computer Vision and Pattern Recognition Workshop, (NTIRE 2018)

Dependencies

• Python 3.5+
• TensorFlow 1.8

Train_Model

• The Installation completely the same as our dependencies. Make sure you have correctly installed before using our code.
• Download the align & cropped version of CelebA dataset for training and testing
• Preprocess the training face images, including detection, alignment, etc. Here we strongly recommend MTCNN, which is an effective and efficient open-source tool for face detection and alignment.
• Put aligned images under "./data/CelebA"
• For L2 version : bash train.sh #NUM_GPU, For L2 + GAN version : bash train_gan.sh #NUM_GPU

Inference_Model

• bash test.sh #NUM_GPU
• For evaluation : run test_psnr.m & test_ssim.m on Matlab

To Do

• Add auxilary classifier loss
• Replace BN in discriminator with SN