Awesome
Exemplar_GAN_Eye_Inpainting
The tensorflow implement of Eye In-Painting with Exemplar Generative Adversarial Networks
Introduction
Network Architecture
<p align="center"> <img src="/images/net.jpg"> </p>Paper result
<p align="center"> <img src="/images/paper_result.jpg"> </p>Noted: Differences with the original paper.
-
Because the origin paper does't provide the details of model, this implement uses the architecture and hyperparamters from SG-GAN(Using adapted residual image learning for G and spectral norm for D)
-
Just use refernece image as the exemplar, not code.
-
Our model trained using 256x256 pixels, not 128x128 mentioned in the original paper.
Dependencies
Usage
- Clone this repo:
git clone https://github.com/zhangqianhui/Exemplar_GAN_Eye_Inpainting.git
- Download the CelebA-ID dataset
You can download CelebA-ID Benchmark dataset according to the Dataset
and unzip CelebA-ID into a directory.
- Train the model using the default parameter
python main.py --OPER_FLAG=0 --path your_path
- Test the model
python main.py --OPER_FLAG=1 --path your_path --test_step= your model_name