Attention Heatmap Visualization

[yimingz1218]

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?

[VlSomers]

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.