[REG] token’s attention score on visual tokens

[5Jiahe]

Can you provide the code for the part that visualizes the [REG] token's attention score on visual tokens in Fig. 3 and Fig. 4, thanks!

[djiajunustc]

Please refer to the tutorial of DETR: https://colab.research.google.com/github/facebookresearch/detr/blob/colab/notebooks/detr_attention.ipynb