Giters

VlSomers / bpbreid

A strong baseline for body part-based person re-identification (check out our WACV23 paper)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Attention Heatmap Visualization

yimingz1218 opened this issue · comments

commented

Hello, thank you for your amazing work! How can I visualize the attention heatmap for different body parts as well as the whole body? I think it should be under the path "torchreid/utils/visualization", but the code is a little bit hard to understand.. Can you offer me some guidances of doing so?

commented

Hi @yimingz1218 , sorry for answering this late, have you found an answer to your question? If you run the codebase and set the "test.visrank" value to True, an image should be created on your disk with the visualizations. Let me know if you manage to create such image.