Skip to content
/ unet Public

"pip install unet": PyTorch Implementation of 1D, 2D and 3D U-Net architecture.

License

Notifications You must be signed in to change notification settings

fepegar/unet

Repository files navigation

U-Net

DOI License

PyTorch implementation of 1D, 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:

Pérez-García, Fernando. (2020). fepegar/unet: PyTorch implementation of 2D and 3D U-Net (v0.7.5). Zenodo. https://doi.org/10.5281/zenodo.3697931