Thank you for sharing with us the great work, it is impressive.
Is there a easy way to visualize attention for custom images? For example, in ./viz/model_view_vl-bert_coco.ipynb, you show attention visualization for coco images, what if I want to use my own images? It does not seem easy to generate attention map through vis_attention_maps.py for a single image.
Dear @huangyz0918 and @jackroos ,
Thank you for sharing with us the great work, it is impressive.
Is there a easy way to visualize attention for custom images? For example, in ./viz/model_view_vl-bert_coco.ipynb, you show attention visualization for coco images, what if I want to use my own images? It does not seem easy to generate attention map through vis_attention_maps.py for a single image.
Thank you for your help.
Best Wishes,
Alex