How to visualize the attention maps like in Figure 6?

[sithu31296]

I want to know how to visualize the attention maps of the decoder layer. Which script do you use?

[kaixinbear]

The visualization code hasn't been released yet. But you can visualize the attention maps in the following steps:

1. Run test script and save the middle variables that you want to visualize such as attention maps as .npy files.
2. Use python lib like matplotlib and opencv to plot feature maps. In this step, you can refer to code for visualization.

[sithu31296]

Thank you. I will try that.