composer.models.unet.unet#

composer.models.unet.unet

Classes

ComposerModel

The minimal interface needed to use a model with composer.trainer.Trainer.

Dice

The Dice Coefficient for evaluating image segmentation.

Tensor

composer.models.unet.unet.torch.Tensor

UNet

A U-Net model extending ComposerClassifier.

UNet

composer.models.unet.model.UNet

Attributes

  • Any

  • BatchPair

  • Metrics

  • Optional

  • Tensors

  • Tuple

  • log

class composer.models.unet.unet.UNet[source]#

Bases: composer.models.base.ComposerModel

A U-Net model extending ComposerClassifier.

See this paper for details on the U-Net architecture.

inference2d(image)[source]#

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