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

Navigation

Verified details

These details have been verified by PyPI
Maintainers
Avatar for fepegar from gravatar.com fepegar

Unverified details

These details have not been verified by PyPI
Project links
Meta
Classifiers

Project description

DOI License https://img.shields.io/pypi/v/unet.svg

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

Supported by

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