how can i get the attention map in the article

Vegetebird / StridedTransformer-Pose3D

[TMM 2022] Exploiting Temporal Contexts with Strided Transformer for 3D Human Pose Estimation

MIT License

335 stars 37 forks source link

how can i get the attention map in the article #42

Closed sayoko17 closed 7 months ago

sayoko17 commented 8 months ago

StridedTransformer-Pose3D/common /plot_att.py i learned about it but i don't understand the code about data_names = sorted(glob.glob(os.path.join('data/', '*.npz'))) ,and i can’t get the attention map in the article

Vegetebird commented 7 months ago

You need to run the model and save the p_attn as an npz file.