tumourkit.classification.infer.load_saved_model

tumourkit.classification.infer.load_saved_model(weights_path: str, conf_path: str, num_classes: int, num_feats: int, enable_background: bool) → Module

Loads a saved model from the given weights_path and conf_path.

Parameters:

• weights_path (str) – The path to the saved weights file.

• conf_path (str) – The path to the configuration file.

• num_classes (int) – The number of classes for the model.

• num_feats (int) – The number of features for the model.

• enable_background (bool) – Enable when model has extra head to correct extra cells.

Returns:

The loaded model.

Return type:

nn.Module