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.mapde

Defines MapDe architecture.

tiatoolbox.models.architecture.micronet

Defines MicroNet architecture.

tiatoolbox.models.architecture.nuclick

Defines original NuClick architecture

tiatoolbox.models.architecture.sccnn

Defines SCCNN architecture.

tiatoolbox.models.architecture.unet

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

tiatoolbox.models.architecture.utils

Defines utility layers and operators for models in tiatoolbox.

tiatoolbox.models.architecture.vanilla

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