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.
pip install unet
If you used this code for your research, please cite this repository using the information available on its Zenodo entry.