FrEIA

Framework for Easily Invertible Architectures

Dependencies

• torch>=0.4.1
• numpy>=1.15.0

The examples additionally require:

• matplotlib>=2.0.0
• tqdm

Installation

Install pytorch>=0.4.1 as described on their website, then install FrEIA using pip:

pip install git+https://github.com/VLL-HD/FrEIA.git

Development

Clone the git repo

git clone https://github.com/VLL-HD/FrEIA.git

then install it in "Development Mode", so that changes don't require a reinstall

cd FrEIA
python setup.py develop

Documentation

Documentation can be found here. Also check out the usage example.

Cite Us

@article{ardizzone2018analyzing,
  title={Analyzing Inverse Problems with Invertible Neural Networks},
  author={Ardizzone, Lynton and Kruse, Jakob and Wirkert, Sebastian and Rahner, Daniel and Pellegrini, Eric W and Klessen, Ralf S and Maier-Hein, Lena and Rother, Carsten and K{\"o}the, Ullrich},
  journal={arXiv preprint arXiv:1808.04730},
  year={2018}
}