Skip to content
No description, website, or topics provided.
Python MATLAB
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.
Demo_mat upload codes Jun 10, 2019
Pretrained_models upload codes Jun 10, 2019
model
training
utils upload codes Jun 10, 2019
README.md Update README.md Jun 24, 2019
pnp_admm_csmri.py upload codes Jun 10, 2019
pnp_admm_photon_imaging.py upload codes Jun 10, 2019
pnp_admm_poisson_denoise.py upload codes Jun 10, 2019
pnp_fbs_csmri.py upload codes Jun 10, 2019
pnp_fbs_poisson_denoise.py upload codes Jun 10, 2019

README.md

Provable_Plug_and_Play

The implement of the following paper:

E. K. Ryu, J. Liu, S. Wang, X. Chen, Z. Wang, and W. Yin. "Plug-and-Play Methods Provably Converge with Properly Trained Denoisers." ICML, 2019.

Scripts

  1. pnp_admm_csmri.py (CS-MRI solved with Plug-and-Play ADMM)
  2. pnp_fbs_csmri.py (CS-MRI solved with Plug-and-Play FBS)
  3. pnp_admm_poisson_denoise.py (Poisson Denoising solved with Plug-and-Play ADMM)
  4. pnp_fbs_poisson_denoise.py (Poisson Denoising solved with Plug-and-Play FBS)
  5. pnp_admm_photon_imaging.py (Single Photon Imaging solved with Plug-and-Play ADMM)
  6. pnp-fbs_photon_imaging.py (to appear soon)
  7. training/train_full_realsn.py (Training the denoisers)

How to run the scripts

Run with default settings

$ python3 pnp_admm_csmri.py

Run with costmized settings

$ python3 pnp_admm_csmri.py --model_type DnCNN --sigma 15 --alpha 2.0 --maxitr 100 --verbose 1

All the arguments are explained in the file "utils/config.py".

Training

We provide some pretraining models in the folder "Pretrained_models". They can be directly used in the Plug-and-PLay framework. To train new models, please refer the "README" file in the "training" folder.

Citation

If you find our code helpful in your resarch or work, please cite our paper.

@InProceedings{pmlr-v97-ryu19a,
  title = 	 {Plug-and-Play Methods Provably Converge with Properly Trained Denoisers},
  author = 	 {Ryu, Ernest and Liu, Jialin and Wang, Sicheng and Chen, Xiaohan and Wang, Zhangyang and Yin, Wotao},
  booktitle = 	 {Proceedings of the 36th International Conference on Machine Learning},
  pages = 	 {5546--5557},
  year = 	 {2019},
  editor = 	 {Chaudhuri, Kamalika and Salakhutdinov, Ruslan},
  volume = 	 {97},
  series = 	 {Proceedings of Machine Learning Research},
  address = 	 {Long Beach, California, USA},
  month = 	 {09--15 Jun},
  publisher = 	 {PMLR},
  pdf = 	 {http://proceedings.mlr.press/v97/ryu19a/ryu19a.pdf},
  url = 	 {http://proceedings.mlr.press/v97/ryu19a.html}
}
You can’t perform that action at this time.