unet#
A U-Net model extending ComposerModel
.
Classes
A U-Net model extending |
- class composer.models.unet.unet.UNet(num_classes=3)[source]#
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
.