Toggle Light / Dark / Auto color theme

Toggle table of contents sidebar

anomalib.utils.callbacks.openvino

Callback that compresses a trained model by first exporting to .onnx format, and then converting to OpenVINO IR.

Module Contents

Classes

OpenVINOCallback

Callback to compresses a trained model.

Attributes

logger

anomalib.utils.callbacks.openvino.logger

class anomalib.utils.callbacks.openvino.OpenVINOCallback(input_size: Tuple[int, int], dirpath: str, filename: str)

Bases: pytorch_lightning.Callback

Callback to compresses a trained model.

Model is first exported to .onnx format, and then converted to OpenVINO IR.

Parameters

• input_size (Tuple[int, int]) – Tuple of image height, width

• dirpath (str) – Path for model output

• filename (str) – Name of output model

on_train_end(trainer, pl_module: anomalib.models.components.AnomalyModule) → None

Call when the train ends.

Converts the model to onnx format and then calls OpenVINO’s model optimizer to get the .xml and .bin IR files.