composer.models.unet#

Modules

`composer.models.unet.model`	The Unet architecture used in image segmentation.
`composer.models.unet.unet`	A U-Net model extending `ComposerModel`.
`composer.models.unet.unet_hparams`	YAHP interface for `UNet`.

The Unet architecture used in image segmentation. The example we are using is for BRATS medical brain tumor dataset.

See the Model Card for more details.

Classes

A U-Net model extending ComposerModel.

Hparams

These classes are used with yahp for YAML-based configuration.

YAHP interface for UNet.