技术栈
dinov2
fuhanghang@yeah.net
18 天前
transformer
·
热力图
·
vit
·
注意力热图
·
dinov2
如何使用自己训练的模型生成注意力热图（模型为DINOv2，基于ViT架构）
搞了整整3天，网上的代码试了很多，生成的都是奇怪的二维热图，终于成功，记录一下心路历程 要提前安装好gradCam包 在网上找的代码，生成的热图都是线性的，很奇怪，把代码贴在下面：