anomalib.utils.callbacks.visualizer.visualizer_image

Image Visualizer Callback.

Module Contents

Classes

ImageVisualizerCallback

Callback that visualizes the inference results of a model.
class anomalib.utils.callbacks.visualizer.visualizer_image.ImageVisualizerCallback(task: str, mode: str, image_save_path: str, inputs_are_normalized: bool = True, show_images: bool = False, log_images: bool = True, save_images: bool = True)[source]

Bases: anomalib.utils.callbacks.visualizer.visualizer_base.BaseVisualizerCallback

Callback that visualizes the inference results of a model.

The callback generates a figure showing the original image, the ground truth segmentation mask, the predicted error heat map, and the predicted segmentation mask.

To save the images to the filesystem, add the ‘local’ keyword to the project.log_images_to parameter in the config.yaml file.

on_predict_batch_end(_trainer: pytorch_lightning.Trainer, _pl_module: anomalib.models.components.AnomalyModule, outputs: Optional[pytorch_lightning.utilities.types.STEP_OUTPUT], _batch: Any, _batch_idx: int, _dataloader_idx: int) → None[source]

Show images at the end of every batch.

Parameters

  • _trainer (Trainer) – Pytorch lightning trainer object (unused).

  • _pl_module (LightningModule) – Lightning modules derived from BaseAnomalyLightning object as

  • images. (currently only they support logging) –

  • outputs (Dict[str, Any]) – Outputs of the current test step.

  • _batch (Any) – Input batch of the current test step (unused).

  • _batch_idx (int) – Index of the current test batch (unused).

  • _dataloader_idx (int) – Index of the dataloader that yielded the current batch (unused).

on_test_batch_end(trainer: pytorch_lightning.Trainer, pl_module: anomalib.models.components.AnomalyModule, outputs: Optional[pytorch_lightning.utilities.types.STEP_OUTPUT], _batch: Any, _batch_idx: int, _dataloader_idx: int) → None[source]

Log images at the end of every batch.

Parameters

  • trainer (Trainer) – Pytorch lightning trainer object (unused).

  • pl_module (LightningModule) – Lightning modules derived from BaseAnomalyLightning object as

  • images. (currently only they support logging) –

  • outputs (Dict[str, Any]) – Outputs of the current test step.

  • _batch (Any) – Input batch of the current test step (unused).

  • _batch_idx (int) – Index of the current test batch (unused).

  • _dataloader_idx (int) – Index of the dataloader that yielded the current batch (unused).