https://github.com/cralji/RFF-Nerve-UTP/blob/main/UNET-Nerve-UTP.ipynb

image_segmentation.models.baseline_unet.kernel_initializer(seed)

image_segmentation.models.baseline_unet.unet_baseline(input_shape=(128, 128, 3), name='UNET', out_channels=1, out_ActFunction='sigmoid')