braindecode.modules.ECA#

class braindecode.modules.ECA(in_channels, kernel_size)[source]#

Efficient Channel Attention [Wang2021].

Parameters:

in_channels (int) – number of input feature channels
kernel_size (int) – kernel size of convolutional layer, determines degree of channel interaction, must be odd.

References

Wang, Q. et al., 2021. ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks. CVPR 2021.

Methods

forward(x)[source]#

Apply the Efficient Channel Attention block to the input tensor.