Skip to content
master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
src
 
 
 
 
 
 
 
 

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 .

About

Code for the paper: Solving Linear Inverse Problems using GAN priors

Resources

License

Releases

No releases published

Packages

No packages published