search

LeapLabTHU / FLatten-Transformer

Official repository of FLatten Transformer (ICCV2023)
377 stars 21 forks source link

关于可视化的一些问题 #10

Closed GloamXun closed 10 months ago

GloamXun commented 10 months ago

论文中出现的DeiT-tiny的Attention Map的可视化是如何完成的，想问一下有没有相关的代码可以参考一下，感谢 1

tian-qing001 commented 10 months ago

Hi @GloamXun, we provide the code for visualization right here. You can use the get_attn_matrix function to obtain the attention map. Hoping that it could address your needs.