Model module

Source code | API

This module contains the declaration of the U-Nets in Keras / Tensorflow as well as the loss functions we used to train them. See our publications for descriptions of the models and losses.

For the seminal paper describing the structure of the U-Net models:

Ronneberger, Olaf, Philipp Fischer, and Thomas Brox. “U-net: Convolutional networks for biomedical image segmentation.” International Conference on Medical image computing and computer-assisted intervention. 2015

Original implementation of U-Net in Tensorflow / Keras:

https://github.com/zhixuhao/unet/blob/master/model.py