odeon.nn.unet.LightUNet

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

LightUnet = U-Net with reduced number of filters

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)