AGA-GAN: Attribute Guided Attention Generative Adversarial Network with U-Net for Face Hallucination
This repository provides code for our paper "AGA-GAN: Attribute Guided Attention Generative Adversarial Network with U-Net for Face Hallucination" (arxiv version)
In this paper, we propose AGA-GAN(Attribute Guided Attention Generative Adversarial Network) which uses a novel attention mechanism to identify the spatial location of key facial attributes and focus the generation process to successfully recover salient facial elements. The AGA-GAN allows the attribute descriptors to learn their spatial position in the image domain using the attribute stream. The attribute streams interact with the main generation stream to allow feature maps carrying information about several facial elements to be incorporated in the final HR prediction. Attention maps generated by the attribute stream enable the main generation stream to focus on the spatial locations of essential features and also provide additional information about the description of facial elements. Feedback from the main generation stream allows the attribute stream to generate attention maps progressively focusing on various facial structures. Consequently, the super-resolved (SR) prediction possesses a high degree of fidelity and enjoys high-frequency details demonstrating superior visual quality. Apart from that, we design a spatial and channel attention-based U-Net for enhancing the visual quality by refinement of facial features and even rectification of visual artifacts present in SR prediction by AGA-GAN. It can be noted that apart from demonstrating a high degree of fidelity, the use of the U-Net module successively after AGA-GAN improves the quantitative performance
1.) make directory named "celebA_128"
2.) Align the faces using mtcnn
Model architecture is defined in aagan_8x.py
Train AGA-GAN as:
python aagan_8x.py
Test using:
python agagan_test.py
For training and testing the AGA-GAN+Unet run:
python agagan_unet.py
Please cite our paper if you find the work useful:
@article{srivastava2021aga,
title={AGA-GAN: Attribute Guided Attention Generative Adversarial Network with U-Net for Face Hallucination},
author={Srivastava, Abhishek and Chanda, Sukalpa and Pal, Umapada},
journal={arXiv preprint arXiv:2111.10591},
year={2021}
}
Please feel free to contact me if you need any advice or guidance in using this work (E-mail)