Open 5Jiahe opened 6 months ago
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!
Please refer to the tutorial of DETR: https://colab.research.google.com/github/facebookresearch/detr/blob/colab/notebooks/detr_attention.ipynb
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!