/SE_attention

Implementation of Squeeze and Excitation attention mechanism applied in Pytorch and in Keras.

Primary LanguagePython

Watchers