Learn to pay attention
We propose an end-to-end-trainable attention module for convolutional neural network (CNN) architectures built for image classification. The module takes as input the 2D feature vector maps which form the intermediate representations of the input image at different stages in the CNN pipeline, and ou...
| Main Authors: | , , , |
|---|---|
| Format: | Conference item |
| Published: |
International Conference on Learning Representations
2018
|