grid_attention_visualization

attention map visualization

pre-requisite

$ pip install opencv-python
$ pip install matplotlib
$ pip install Pillow
$ pip install numpy

Contents

• Version1 is here visualize_attention_map.py
• Version2 (recommended) is here visualize_attention_map_V2.py

the attention style in version2 can be changed by "cmap", choose the color map you like here

Result

1. Version1 example

• original image

• image with attention (for example we use random attention map here)

2. Version2 example

• original image

• image with attention (for example we use random attention map here)