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.

Files for unet, version 0.7.7
Filename, size File type Python version Upload date Hashes
Filename, size unet-0.7.7-py2.py3-none-any.whl (8.1 kB) File type Wheel Python version py2.py3 Upload date Hashes View
Filename, size unet-0.7.7.tar.gz (12.3 kB) File type Source Python version None Upload date Hashes View

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring DigiCert DigiCert EV certificate Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page