search

yangsenius / TransPose

PyTorch Implementation for "TransPose: Keypoint localization via Transformer", ICCV 2021.
https://github.com/yangsenius/TransPose/releases/download/paper/transpose.pdf
MIT License
360 stars 58 forks source link

Code for visualization #8

Closed weixuansun closed 3 years ago

weixuansun commented 3 years ago

Thanks for the great work, I wonder can you please share the code of how you visualize the attention map? or did I miss it somewhere?

yangsenius commented 3 years ago

Thanks for your interest~ The code for visualizing the attention maps is not included in the current repo. We may share it in the future.

weixuansun commented 3 years ago

Thanks for your interest~ The code for visualizing the attention maps is not included in the current repo. We may share it in the future.

Thanks, looking forward to it.