braindecode.modules.EncNet#

class braindecode.modules.EncNet(in_channels: int, n_codewords: int)[source]#

Context Encoding for Semantic Segmentation from [Zhang2018].

Parameters:

• in_channels (int) – number of input feature channels

• n_codewords (int) – number of codewords

References

[Zhang2018]

Zhang, H. et al. 2018.

Context Encoding for Semantic Segmentation. CVPR 2018.

Methods

forward(x)[source]#

Apply attention from the Context Encoding for Semantic Segmentation.

Parameters:

x (Pytorch.Tensor)

Return type:

Pytorch.Tensor