replace_strides_with_dilation

tiatoolbox . models . architecture . timm_efficientnet . replace_strides_with_dilation

replace_strides_with_dilation(module, dilation_rate)

Replace strides with dilation in Conv2d layers.

Converts convolutional layers to use dilation instead of stride, enabling atrous convolutions for semantic segmentation tasks.

Parameters:

• module (nn.Module) – Module containing Conv2d layers to patch.

• dilation_rate (int) – Dilation rate to apply to all Conv2d layers.

Return type:

None

Example

>>> replace_strides_with_dilation(model, dilation_rate=2)