DSCBlock

class eyefeatures.deep.models.DSCBlock(in_channels, out_channels, kernel_size=3, stride=1, padding=1)[source]

Bases: Module

Depthwise separable convolution, which consists of a depthwise convolution followed by a pointwise convolution.

Parameters:

  • in_channels – (int) Number of input channels.

  • out_channels – (int) Number of output channels.

  • kernel_size – (int, optional) Size of the convolution kernel. Default is 3.

  • stride – (int, optional) Stride for the convolution layers. Default is 1.

  • padding – (int, optional) Padding for the convolution layers. Default is 1.

Returns:

Output tensor after applying depthwise and pointwise convolutions, batch normalization, and ReLU activation.