U-Net

.. image:: https://zenodo.org/badge/DOI/10.5281/zenodo.3522306.svg :target: https://doi.org/10.5281/zenodo.3522306 :alt: DOI

.. image:: https://img.shields.io/badge/License-MIT-yellow.svg :target: https://opensource.org/licenses/MIT :alt: License

.. image:: https://img.shields.io/pypi/v/unet.svg :target: https://pypi.python.org/pypi/unet

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 <https://arxiv.org/abs/1505.04597>. The 3D version was described in Çiçek et al. 2016, 3D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotation <https://arxiv.org/abs/1606.06650>.

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 <https://doi.org/10.5281/zenodo.3697931>_:

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