/unet-pytorch

Primary LanguagePythonMIT LicenseMIT

3D U-Net in PyTorch

3D version of UNet from publication::
Çiçek Ö, Abdulkadir A, Lienkamp SS, Brox T, Ronneberger O. 3D U-Net: learning dense volumetric segmentation from sparse annotation. In International Conference on Medical Image Computing and Computer-Assisted Intervention 2016 Oct 17 (pp. 424-432). Springer International Publishing.

Implemented in Python 3.6 and PyTorch 0.3.0.

Copyright (c) 2018 Charles Houston