/MA_LearnedOperatorCorrectionISTA

Learned operator correction for the proximal gradient method of the Master thesis

Primary LanguagePython

MA_LearnedOperatorCorrectionISTA

The Parts of the code I used for the results of my Master's thesis "Learned operator correction for the proximal gradient"

necessary Libraries

the following packages can be installed with pip

  • scipy
  • numpy
  • matplotlib
  • torchinfo
  • dival
  • tqdm

I recommend anaconda for

and the following libary need to be installed directly

content

  • one can create a schift dataset using 'create_shift_dataset'
  • the jupyter-notebook 'Training' was used to setup a config file with 'conf' and start the training using 'Setup_and_training' (not recommended to do that on a gpu)
  • the jupyter-notebook 'Pictures for master thesis' was used to create the plots in the numerical part of the thesis (can take long on just a Cpu)