Visualization of word-image attention maps

Thank you for great work.

Would you provide a source code or some implementation details for visualization of word-image attention maps used to plot Figure 4 and Figure A3 in the paper?

Especially, how was K-means clustering applied?