SCSEModule

tiatoolbox . models . architecture . utils . SCSEModule

class SCSEModule(in_channels, reduction=16)

Spatial and Channel Squeeze & Excitation (SCSE) module.

Initialize the SCSE module.

Parameters:

• in_channels (int) – Number of input channels.

• reduction (int) – Reduction ratio for channel attention.

Methods

forward

Forward pass of the SCSE module.

Attributes

training

forward(x)

Forward pass of the SCSE module.

Parameters:

• x (torch.Tensor) – Input feature map of shape (N, C, H, W).

• self (SCSEModule)

Returns:

Output feature map after applying SCSE attention.

Return type:

torch.Tensor