NVIDIA
UNet Medical for TensorFlow2
Resource
NVIDIA
UNet Medical for TensorFlow2

U-Net allows for seamless segmentation of 2D images, with high accuracy and performance.

Changelog

February 2020

  • Initial release

Known issues

  • For TensorFlow 2.0 the training performance using AMP and XLA is around 30% lower than reported here. The issue was solved in TensorFlow 2.1.
  • Due to random initialization, around 5% of training runs result in lower DICE score, usually around 0.81.