frankzhangyk / cnn-rpgd Goto Github PK

==================================
New announcement: For a python/pytorch based version please refer to https://github.com/PhanHuyThong/Image-Reconstruction-by-CNN-based-PGD/

In that repo the pytorch code is used in training the projector network. This trained network can then be plugged in RPGD scheme which has both python and matlab support. For the matlab case onnx is used to translate the learnt network from pytorch to matlab. The parameters for the projector training and RPGD can be passed on using config file. The forward model that was used in [1] was the radon/iradon of matlab. However, similar forward model can be found in python.
I highly recommend using the code from that repo.
==================================


This repo is in matconvnet.




[1] H. Gupta, K. H. Jin, H.Q.Nguyen, M.T. McCann, and M. Unser, 'CNN-Based Projected Gradient Descent for Consistent Image Reconstruction', IEEE TMI, 2018. https://ieeexplore.ieee.org/abstract/document/8353870

[2]  K. H. Jin, M.T. McCann, E. Froustey, and M. Unser, 'Deep CNN for Inverse problem in Imaging', IEEE Transactions on Image Processing, 2017. http://ieeexplore.ieee.org/abstract/document/7949028/

Readme

1. Before launching CNN-RPGD, kindly install the MatConvNet (http://www.vlfeat.org/matconvnet/) (For the GPU, it needs a different compilation.)
2. Modify addpathsRPGD and addpathsPT based on the realtive directory paths on your machine. TrainingCTMeasurementModel, RPGDCTMeasurementModel should be edited to change the default parameters of RPGD.
3. The codes TrainingCTMeasurementModel.m is based on GPU computation. Kindly modify it to use it on CPU.
4. Use TrainingCTMeasurementModel for training the projector. After training, run RPGDCTMeasurementModel to get the result on the test data.

*note : these codes successfully ran on Matlab 2016a with GPU TITAN X (architecture : Maxwell)
contact : Harshit Gupta ([email protected]),