/attention_mechanism_cnn

An implementation of an attention mechanism for CNNs. (Based on the paper: Rodriguez et al. ECCV2018)

Primary LanguagePython

attention_mechanism_cnn

An implementation of a modular attention mechanism for Convolutional Neural Networks.

It is largely based on the paper:

Rodríguez, Pau, et al. "Attend and rectify: a gated attention mechanism for fine-grained recovery." Proceedings of the European Conference on Computer Vision (ECCV). 2018.