How to get the learned superpixel tokens?

[Zhaoyi-Yan]

https://github.com/hhb072/STViT/blob/main/models/stvit.py#L249, is pixel_features the learned superpixel tokens? You have already visualized the attention maps, however, how to visualize the learned tokens? Thank you for your help!

[hhb072]

Thank you for the reminder. We have uploaded the code for visualizing super tokens, named visualize_super_tokens.py, to the visualization folder.