visualization

ViT

参考 https://github.com/jeonsworld/ViT-pytorch/blob/main/visualize_attention_map.ipynb , 该示例代码拷贝在ViT/visualize_attention_map.ipynb

CNN

用到的的工具是pytorch_grad_cam , 示例代码在 CNN/example.py

其他参考链接

ViT: https://nbviewer.org/github/luo3300612/Visualizer/blob/main/demo.ipynb

CNN: https://zhuanlan.zhihu.com/p/438234233