ai-med / squeeze_and_excitation

PyTorch Implementation of 2D and 3D 'squeeze and excitation' blocks for Fully Convolutional Neural Networks
https://arxiv.org/abs/1808.08127
MIT License
312 stars 52 forks source link