Neural network layers.

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

Bases: Module

SSPCAB block.

  • 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.

Initializes internal Module state, shared by both nn.Module and ScriptModule.

forward(inputs: Tensor) Tensor[source]

Forward pass through the SSPCAB block.

training: bool