unet¶
tiatoolbox
.models
.architecture
.unet
Define a set of UNet variants to be used within tiatoolbox.
Functions
Helper to create a block of Vanilla Convolution. |
Classes
A subclass of ResNet defined in torch. |
|
Generate families of UNet model. |
|
Construct a basic UNet encoder. |