UNetModel

class aydin.nn.models.unet.UNetModel(spacetime_ndim, nb_unet_levels: int = 4, nb_filters: int = 8, learning_rate=0.01, pooling_mode: str = 'max')

forward(x)

UNet forward method.

Parameters

x

input_masknumpy.ArrayLike

A mask per image must be passed with self-supervised training.