Giters

zengyan-97 / X-VLM

X-VLM: Multi-Grained Vision Language Pre-Training (ICML 2022)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Drawing Attention Heatmap

TheodorPatrickZ opened this issue · comments

commented

Hello,

how did you draw the attention Heatmaps in Figure 3 of your paper?

commented

Hi,
I have not used grad-cam for a long time.
You can refer to this code: https://github.com/salesforce/ALBEF/blob/main/visualization.ipynb for grad-cam.