/pytorch_generic_unet

Generic implementation of pytorch basic and nested unet.

Primary LanguagePythonMIT LicenseMIT

ReadMe

Generic implementation of pytorch basic and nested unet. Certain internal implementation (e.g., hooks) relies on fastai.

Basic Usage

Basic UNet with customizable encoder path:

from generic_unet import BasicUNet

Nested UNet with customizable encoder and number of nested levels.

from generic_unet import NestedUNet

TODO

Detailed showcase and full documentation.