Question about the vtisualizations of the selected tokens at each HTI module

[duduxx123]

Hello! Thank you for your great work! I would like to know how the visualization of the selected tokens in each HTI module on the COCO dataset, as shown in Figure 5 of the paper, was achieved? Thank you again!