UNet¶

class torchelie.models.UNet(arch: List[int], num_classes: int)¶

Experimental: U-Net from U-Net: Convolutional Networks for Biomedical Image Segmentation. This net has architectural changes operations for further customization.

Parameters

arch (List[int]) – a list of channels from the outermost to innermost layers
num_classes (int) – number of output channels

Warning

UNet() is experimental, and may change or be deleted soon if not already broken