How to draw the attention map of ViTSTR?

roatienza / deep-text-recognition-benchmark

PyTorch code of my ICDAR 2021 paper Vision Transformer for Fast and Efficient Scene Text Recognition (ViTSTR)

Apache License 2.0

284 stars 57 forks source link

How to draw the attention map of ViTSTR? #24

Closed lexiaoyuan closed 2 years ago

lexiaoyuan commented 2 years ago

Hello, thank you very much for being able to open the source code, it is a very rewarding work. I am a graduate student and I want to do a little bit of my own experiment based on ViTSTR. Now I want to draw an attention map similar to the one shown in Fig. 9 of your paper, can you give me some help? Thank you very much!

roatienza commented 2 years ago

Hi, Attention rollout: https://github.com/jacobgil/vit-explain