PyTorch implementation of 2D and 3D U-Net
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.
- First release on PyPI.
Release history Release notifications | RSS feed
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
|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|