Model module¶
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:
Original implementation of U-Net in Tensorflow / Keras: