monai.deploy.core.models.ModelFactory#

class monai.deploy.core.models.ModelFactory[source]#

Bases: object

ModelFactory is a class that provides a way to create a model object.

Methods

__init__()

create(path[, name, model_type])

Creates a model object.

detect_model_type(path[, model_type])

Detects the model type based on a model path.
static create(path, name='', model_type='')[source]#

Creates a model object.

Parameters:

  • path (Union[str, Path]) – A path to the model.

  • name (str) – A name of the model.

  • model_type (str) – A type of the model.

Return type:

Optional[Model]

Returns:

A model object. Returns None if the model file/folder does not exist.

static detect_model_type(path, model_type='')[source]#

Detects the model type based on a model path.

Parameters:

  • path (Union[str, Path]) – A path to the model file/folder.

  • model_type (str) – A model type.

Return type:

Tuple[str, Optional[Type[Model]]]

Returns:

A tuple of the model type string and the model class.