NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
nnU-Net for PyTorch
Resource
NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
nnU-Net for PyTorch

An optimized, robust and self-adapting framework for U-Net based medical image segmentation

Changelog

December 2021

  • Container updated to 21.11
  • Use MONAI DynUNet instead of custom U-Net implementation
  • Add balanced multi-gpu evaluation
  • Support for evaluation with resampled volumes to original shape

October 2021

  • Add Jupyter Notebook with BraTS solution

May 2021

  • Add Triton Inference Server support
  • Removed deep supervision, attention and drop block

March 2021

  • Container updated to 21.02
  • Change data format from tfrecord to npy and data loading for 2D

January 2021

  • Initial release
  • Add notebook with custom dataset loading

Known issues

There are no known issues in this release.