anomalib.models.components.layers

Neural network layers.

Submodules

Package Contents

Classes

SSPCAB

SSPCAB block.
class anomalib.models.components.layers.SSPCAB(in_channels: int, kernel_size: int = 1, dilation: int = 1, reduction_ratio: int = 8)[source]

Bases: torch.nn.Module

SSPCAB block.

Parameters

  • in_channels (int) – Number of input channels.

  • kernel_size (int) – Size of the receptive fields of the masked convolution kernel.

  • dilation (int) – Dilation factor of the masked convolution kernel.

  • reduction_ratio (int) – Reduction ratio of the attention module.

forward(inputs: torch.Tensor) → torch.Tensor

Forward pass through the SSPCAB block.