odeon.nn.unet.UNet

class odeon.nn.unet.UNet(*args: Any, **kwargs: Any)

U-Net class

Parameters

• n_channels (int) – number of input channels

• n_classes (int) – number of output classes

__init__(n_channels, n_classes)

Methods

__init__(n_channels, n_classes)
forward(x)