Tasks#

Task Initialization of OTX Classification.

class otx.algorithms.classification.tasks.ClassificationInferenceTask(task_environment: TaskEnvironment, **kwargs)#

Inference Task Implementation of OTX Classification.

evaluate(output_resultset: ResultSetEntity, evaluation_metric: Optional[str] = None)#: Evaluate function of OTX Classification Task.

explain(dataset: DatasetEntity, explain_parameters: Optional[InferenceParameters] = None) → DatasetEntity#: Main explain function of OTX Classification Task.

export(export_type: ExportType, output_model: ModelEntity, precision: ModelPrecision = ModelPrecision.FP32, dump_features: bool = False)#: Export function of OTX Classification Task.

infer(dataset: DatasetEntity, inference_parameters: Optional[InferenceParameters] = None) → DatasetEntity#: Main infer function of OTX Classification.

unload()#: Unload function of OTX Classification Task.

class otx.algorithms.classification.tasks.ClassificationNNCFTask(task_environment: TaskEnvironment, **kwargs)#: ClassificationNNCFTask.

class otx.algorithms.classification.tasks.ClassificationOpenVINOTask(task_environment: TaskEnvironment)#

Task implementation for OTXClassification using OpenVINO backend.

deploy(output_model: ModelEntity) → None#: Deploy function of ClassificationOpenVINOTask.

evaluate(output_resultset: ResultSetEntity, evaluation_metric: Optional[str] = None)#: Evaluate function of ClassificationOpenVINOTask.

explain(dataset: DatasetEntity, explain_parameters: Optional[InferenceParameters] = None) → DatasetEntity#: Explain function of ClassificationOpenVINOTask.

infer(dataset: DatasetEntity, inference_parameters: Optional[InferenceParameters] = None) → DatasetEntity#: Infer function of ClassificationOpenVINOTask.

load_inferencer() → ClassificationOpenVINOInferencer#: load_inferencer function of ClassificationOpenVINOTask.

optimize(optimization_type: OptimizationType, dataset: DatasetEntity, output_model: ModelEntity, optimization_parameters: Optional[OptimizationParameters] = None)#: Optimize function of ClassificationOpenVINOTask.

class otx.algorithms.classification.tasks.ClassificationTrainTask(task_environment: TaskEnvironment, **kwargs)#

Train Task Implementation of OTX Classification.

cancel_training()#

Cancel training function in ClassificationTrainTask.

Sends a cancel training signal to gracefully stop the optimizer. The signal consists of creating a ‘.stop_training’ file in the current work_dir. The runner checks for this file periodically. The stopping mechanism allows stopping after each iteration, but validation will still be carried out. Stopping will therefore take some time.

save_model(output_model: ModelEntity)#: Save best model weights in ClassificationTrainTask.

train(dataset: DatasetEntity, output_model: ModelEntity, train_parameters: Optional[TrainParameters] = None)#: Train function in ClassificationTrainTask.