anomalib.utils.callbacks.metrics_configuration

Metrics Configuration Callback.

Module Contents

Classes

MetricsConfigurationCallback

Metrics Configuration Callback.
class anomalib.utils.callbacks.metrics_configuration.MetricsConfigurationCallback(adaptive_threshold: bool, default_image_threshold: Optional[float] = None, default_pixel_threshold: Optional[float] = None, image_metric_names: Optional[List[str]] = None, pixel_metric_names: Optional[List[str]] = None, normalization_method: str = 'min_max')[source]

Bases: pytorch_lightning.callbacks.Callback

Metrics Configuration Callback.

setup(self, _trainer: pytorch_lightning.Trainer, pl_module: pytorch_lightning.LightningModule, stage: Optional[str] = None) → None[source]

Setup image and pixel-level AnomalibMetricsCollection within Anomalib Model.

Parameters

  • _trainer (pl.Trainer) – PyTorch Lightning Trainer

  • pl_module (pl.LightningModule) – Anomalib Model that inherits pl LightningModule.

  • stage (Optional[str], optional) – fit, validate, test or predict. Defaults to None.