Layers¶

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.

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.

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

forward(inputs: Tensor) → Tensor[source]¶: Forward pass through the SSPCAB block.

training: bool¶