architectureΒΆ

Defines a set of models to be used within tiatoolbox.

Functions

fetch_pretrained_weights

Get the pretrained model information from yml file.

get_pretrained_model

Load a predefined PyTorch model with the appropriate pretrained weights.

tiatoolbox.models.architecture.hovernet

tiatoolbox.models.architecture.hovernetplus

tiatoolbox.models.architecture.idars

Defines CNNs as used in IDaRS for prediction of molecular pathways and mutations.

tiatoolbox.models.architecture.micronet

Defines MicroNet architecture.

tiatoolbox.models.architecture.unet

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

tiatoolbox.models.architecture.utils

Defines utlity layers and operators for models in tiatoolbox.

tiatoolbox.models.architecture.vanilla

Defines vanilla CNNs with torch backbones, mainly for patch classification.