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.

Models monai.deploy.core.models.Model