Skip to main content

PyTorch implementation of 2D and 3D U-Net

Project description

DOI License

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.


0.4.0 (2019-10-29)

  • First release on PyPI.

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

unet-0.7.7.tar.gz (12.3 kB view hashes)

Uploaded Source

Built Distribution

unet-0.7.7-py2.py3-none-any.whl (8.1 kB view hashes)

Uploaded Python 2 Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page