search

damo-cv / TransReID

[ICCV-2021] TransReID: Transformer-based Object Re-Identification
MIT License
801 stars 175 forks source link

注意力可视化 #47

Open GJTNB opened 2 years ago

GJTNB commented 2 years ago

请问开源代码中是否包括了注意力可视化部分

heshuting555 commented 2 years ago

Currently No.

We refer to the following code base for the visualization code in the paper. https://github.com/jacobgil/vit-explain https://github.com/jeonsworld/ViT-pytorch/blob/main/visualize_attention_map.ipynb

Hope the answer is helpful.

Miaohct commented 9 months ago

你好可以你可以实现注意力的可视化嘛