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.
arch (List[int]) – a list of channels from the outermost to innermost layers
num_classes (int) – number of output channels
UNet() is experimental, and may change or be deleted soon if not already broken