U-Net

PyTorch implementation of 2D and 3D U-Net.

The U-Net architecture was first described in Ronneberger et al. 2015, U-Net: Convolutional Networks for Biomedical Image Segmentation. The 3D version was described in Çiçek et al. 2016, 3D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotation.

Installation

pip install unet

Credits

If you used this code for your research, please cite this repository using the information available on its Zenodo entry.