Fully self-contained source code release including pre-trained checkpoints and a test set for reproducing all figures and experiments in the paper.
The initial release is in PyTorch, with a TensorFlow notebook coming soon.
The main dependencies are
python 3.6.xpytorch (tested with 0.4.1, v0.4.0 is known to not work)numpy,skimage,matplotlib