This repository contains the code for the article "Learning to solve inverse problems using Wasserstein loss".
The code contains the following
- Training using circle phantoms
The pre-trained networks are currently under finalization and will be released soon, in the meantime, training is just a few hours.
The code is currently based on the latest version of ODL and the utility library adler. They can be most easily installed by running
$ pip install https://github.com/odlgroup/odl/archive/master.zip
$ pip install https://github.com/adler-j/adler/archive/master.zip
The learning requires tensorflow, and the ray-transform needs ASTRA for computational feasibility
$ conda install -c astra-toolbox astra-toolbox
Jonas Adler, PhD student
KTH, Royal Institute of Technology
Elekta
jonasadl@kth.se
Axel Ringh, PhD student
KTH, Royal Institute of Technology
aringh@kth.se
Ozan Öktem, Associate Professor
KTH, Royal Institute of Technology
ozan@kth.se
Johan Karlsson, Associate Professor
KTH, Royal Institute of Technology
johan.karlsson@math.kth.se
Development is financially supported by the Swedish Foundation for Strategic Research as part of the project "Low complexity image reconstruction in medical imaging" and "3D reconstruction with simulated forward models".
Development has also been financed by Elekta.