Code for the paper: Solving Linear Inverse Problems using GAN priors
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
celebA_dcgan
exp_scripts update Feb 26, 2018
setup
src Update pgdgan.py Feb 26, 2018
.gitignore
LICENSE
README.md

README.md

Solving Linear Inverse Problems using GANs

Code for the paper: Solving Linear Inverse Problems Using GAN Priors: An Algorithm with Provable Guarantees.

Requirements


To run this code, you require Python 2.7, Tensorflow 1.0.1 (preferably with GPU support), Scipy and PyPNG.

Instructions


  1. Clone the repository, and run all the commands from the parent directory, pgdgan/.

  2. Download the datasets with the script*:

    $ ./setup/download_data.sh 
  3. To train the DCGAN on celebA from scratch, please visit https://github.com/carpedm20/DCGAN-tensorflow, and follow the instructions. Else, pretrained GAN model is available, courtesy Bora et al. To download it, please run the following script*:

    $ ./setup/download_models.sh

    Make sure the model is located at ./models/celebA_64_64/

  4. Run following to run the experiment:

    $ python ./src/pgdgan.py

    You can also use the script available in ./exp_scripts/

* replicated from https://github.com/AshishBora/csgm .