How to visualize the attention map in FFVT?

Markin-Wang / FFVT

[BMVC 2021] The official PyTorch implementation of Feature Fusion Vision Transformer for Fine-Grained Visual Categorization

Other

49 stars 10 forks source link

How to visualize the attention map in FFVT? #1

Closed OhJackHu closed 2 years ago

Markin-Wang commented 2 years ago

Hi, you can refer to the vit-pytorch repository to see how to visualize the attention map via the visualize_attention_map.ipynb file. Generally, you need to keep the attention score matrix and then utilize it to generate the attention map.

OhJackHu commented 2 years ago

十分感谢！