unet

tiatoolbox . models . architecture . unet

Defines a set of ResNet variants to be used within tiatoolbox.

Classes

ResNetEncoder	A subclass of ResNet defined in torch.
UNetModel	Generate families of UNet model.
UnetEncoder	Construct a basic UNet encoder.