Skip to content

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

License

Notifications You must be signed in to change notification settings

weituo12321/pgdgan

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 98.5%
  • Shell 1.5%