PyTorch implementation of 2D and 3D U-Net
Project description
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.
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.
History
0.4.0 (2019-10-29)
First release on PyPI.
Project details
Release history Release notifications | RSS feed
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.6.tar.gz
(12.3 kB
view hashes)