Skip to content

core.services.model.export.model_exporter

model_exporter

Classes:

Name Description
ModelExporter

Base class for exporting and saving a model.

ModelExporter(model)

Base class for exporting and saving a model.

This class provides a standard interface for exporting models and saving them to a Picsellia experiment or model version.

Parameters:

Name Type Description Default

model

Model

The model to export.

required

Methods:

Name Description
export_model

Abstract method to export the model.

save_model_to_experiment

Save exported model to a Picsellia experiment.

save_model_to_model_version

Save exported model to a Picsellia model version.

Attributes:

Name Type Description
model

model = model instance-attribute

export_model(exported_model_destination_path, export_format, hyperparameters) abstractmethod

Abstract method to export the model.

Must be implemented in subclasses.

Parameters:

Name Type Description Default
exported_model_destination_path
str

Directory to export the model to.

required
export_format
str

Format to export the model in.

required
hyperparameters
Any

Optional export configuration.

required

save_model_to_experiment(experiment, exported_weights_path, exported_weights_name)

Save exported model to a Picsellia experiment.

Parameters:

Name Type Description Default
experiment
Experiment

Target experiment.

required
exported_weights_path
str

Path to exported weights directory.

required
exported_weights_name
str

File name to use in Picsellia.

required

save_model_to_model_version(model_version, exported_weights_path, exported_weights_name)

Save exported model to a Picsellia model version.

Parameters:

Name Type Description Default
model_version
ModelVersion

Target model version.

required
exported_weights_path
str

Path to exported weights directory.

required
exported_weights_name
str

File name to use in Picsellia.

required