otx.api.usecases.tasks.interfaces.optimization_interface#

This module contains the interface class for tasks that can optimize their models.

Classes

IOptimizationTask()

A base interface class for tasks which can optimize their models.

OptimizationType(value)

This class enumerates the OPENVINO optimization types.
class otx.api.usecases.tasks.interfaces.optimization_interface.IOptimizationTask[source]#

Bases: object

A base interface class for tasks which can optimize their models.

abstract optimize(optimization_type: OptimizationType, dataset: DatasetEntity, output_model: ModelEntity, optimization_parameters: OptimizationParameters | None)[source]#

This method defines the interface for optimization.

Parameters:

  • optimization_type – The type of optimization

  • dataset – Optional dataset which may be used as part of the optimization process

  • output_model – Output model

  • optimization_parameters – Additional optimization parameters

class otx.api.usecases.tasks.interfaces.optimization_interface.OptimizationType(value)[source]#

Bases: Enum

This class enumerates the OPENVINO optimization types.