An open API service indexing awesome lists of open source software.

https://github.com/fepegar/unet

"pip install unet": PyTorch Implementation of 1D, 2D and 3D U-Net architecture.
https://github.com/fepegar/unet

convolutional-neural-networks deep-learning medical-imaging unet unet-image-segmentation unet-pytorch

Last synced: 2 months ago
JSON representation

"pip install unet": PyTorch Implementation of 1D, 2D and 3D U-Net architecture.

Awesome Lists containing this project

README

        

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 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