Visualize the attention map using heatmap

[chentairan]

Hi GaoChen,

#35
I have the same problem, can you give me some detailed tips

Thanks,
Tairan Chen

[gaochen315]

After obtaining the attention map

iCAN/lib/networks/iCAN_ResNet50_VCOCO.py

Line 320 in 752ce5b

Att_O = self.attention_norm_O(Att_O, 'Norm_Att_O')

You can color code the attention map by O_att = gray2rgb(O_att, cmap='jet'). Then you can either save this map or plot it for visualization.

Let me know if you have any other questions.

[chentairan]

Thanks, I have dealed with it.