PyTorch implementation 3D U-Net and its variants.

The code allows for training the U-Net for both: semantic segmentation (binary and multi-class) and regression problems (e.g. de-noising, learning deconvolutions).

Keywords: Unet, U-net, 3D U-net, semantic segmentation, computer vision, PyTorch, de-noising, deep learning, neural networks

• 3D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotation

• Superhuman Accuracy on the SNEMI3D Connectomics Challenge

• 1
• 1
• 1