composer.models.unet.unet

A U-Net model extending ComposerModel.

Classes

UNet	A U-Net model extending ComposerModel.

class composer.models.unet.unet.UNet(num_classes=3)

Bases: composer.models.base.ComposerModel

A U-Net model extending ComposerModel.

See U-Net: Convolutional Networks for Biomedical Image Segmentation (Ronneberger et al, 2015) on the U-Net architecture.

Parameters

num_classes (int, optional) – The number of classes. Needed for classification tasks. Default: 3.

inference2d(image)

Runs inference on a 3D image, by passing each depth slice through the model.