The need for visualization functions

Yangr116 / ScalableViT

This is code of paper "ScalableViT: Rethinking the Context-oriented Generalization of Vision Transformer"

26 stars 3 forks source link

The need for visualization functions #4

Open whiteBAI-97 opened 1 year ago

whiteBAI-97 commented 1 year ago

Hello. Thank you very much for your dedication. I was reading this article and found that the visualization feature map of the model is displayed very well. I wonder in what way you visualized the feature map, and I hope you can share the code of the visualization function if it is convenient. Thank you very much, and I look forward to your reply.

Yangr116 commented 1 year ago

I mainly referred to this repo to visualize.

First, you should create and load your model by timm;
Next, you need to separate your model by model.children(), so your can control your model by some indices.
Finally, you can get feature maps by running the separated model, and visualize them.

I used plt.imshow(feat_map, cmap='coolwarm') to visualize. You can change the cmap to change the color you enjoyed.

Good Luck~