architecture¶

Define 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

Define HoVerNet architecture.

tiatoolbox.models.architecture.hovernetplus

Define HoVerNetPlus architecture.

tiatoolbox.models.architecture.idars

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

tiatoolbox.models.architecture.mapde

Define MapDe architecture.

tiatoolbox.models.architecture.micronet

Define MicroNet architecture.

tiatoolbox.models.architecture.nuclick

Define original NuClick architecture.

tiatoolbox.models.architecture.sccnn

Define SCCNN architecture.

tiatoolbox.models.architecture.unet

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

tiatoolbox.models.architecture.utils

Define utility layers and operators for models in tiatoolbox.

tiatoolbox.models.architecture.vanilla

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