anomalib.utils.callbacks.tiler_configuration

Tiler Callback.

Module Contents

Classes

TilerConfigurationCallback

Tiler Configuration Callback.
class anomalib.utils.callbacks.tiler_configuration.TilerConfigurationCallback(enable: bool = False, tile_size: Union[int, Sequence] = 256, stride: Optional[Union[int, Sequence]] = None, remove_border_count: int = 0, mode: str = 'padding', tile_count: int = 4)[source]

Bases: pytorch_lightning.callbacks.Callback

Tiler Configuration Callback.

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

Setup Tiler object 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.

Raises

ValueError – When Anomalib Model doesn’t contain Tiler object, it means the model doesn not support tiling operation.