Skip to content

UNet

U-Net is a convolutional neural network (CNN) that's used for image segmentation, particularly in biomedical applications.

import

from atanu_mlkit.models import UNET

in_channels = 3 # Number of channels
num_classes = 5 # Number of classes

model = UNet(in_channels, num_classes)