Skip to content

Invert a pre-trained GAN model (includes code for training a GAN on celebA)

Notifications You must be signed in to change notification settings

ToniCreswell/InvertingGAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Inverting the Generator of a Generative Adversarial Network

Code for reproducing our experiments in: https://arxiv.org/pdf/1802.05701.pdf

To use code:

  1. Download the celebA dataset from here

OR

Download the Shoes dataset from here

OR

Download the Omniglot dataset from here

  1. Install dependencies listed in req.txt

  2. You will also need pyTorch which may be downloaded from here

  3. Run this Jupyter notebook to get the data tensors for CelebA and move them into folder InData/CELEBA/

OR

Run this Jupyter notebook to get the data tensors xShoes.npy and yShoes.npy and move them in to folder InData/SHOES/

  1. The code may be run from cmd line with various options detailed in the code

Example results:

alt text

About

Invert a pre-trained GAN model (includes code for training a GAN on celebA)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published