Suchalongtime/1d_channel-spatial_se_layers

PyTorch implementation of 1D Channel/Spatial Squeeze and Excitation layers for convnets

1D Squeeze and Excitation layers for ConvNets

PyTorch implementation of 1D Channel/Spatial Squeeze and Excitation layers for convnets

---

Code lightly modified from initial 2d/3d implementation on AI Med's repository.

Included

• Channel SE layer
• Spatial SE layer
• Channel+Spatial SE layer

Further info:

For further reading and tests: (i) https://arxiv.org/pdf/1709.01507.pdf
(ii) https://arxiv.org/pdf/1803.02579.pdf
(iii) https://arxiv.org/pdf/1808.08127.pdf