Skip to content

A simple Pytorch implementation of plug and play ADMM with examples

License

Notifications You must be signed in to change notification settings

kanglin755/plug_and_play_admm

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

33 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Plug and Play ADMM

This repository provides a simple Pytorch implementation of plug and play ADMM with examples.

The notebook pnp_admm_example.ipynb contains a example in which a pretrained convnet gaussian denoiser is downloaded (120MB) and plugged into ADMM for solving a motion deblur, inpainting, and super-resolution problem. You can also view in Google Colab.

The notebook denoiser_training.ipynb contains code for training a denoiser from scratch using a subset of ImageNet as trainingset. The trainingset will be download automatically (250MB). You can also view in Google Colab.

Degraded PnP output Ground truth
Motion deblur
Inpainting
Super-resolution

References:
S. Venkatakrishnan, C. Bouman, and B. Wohlberg, “Plug-and-play priors for model based reconstruction,” in Proc. IEEE Global Conference on Signal and Information Processing, 2013, pp. 945–948.

About

A simple Pytorch implementation of plug and play ADMM with examples

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published