可视化工程下载链接
https://download.csdn.net/download/weixin_42899627/76473610
可视化是需要图像和它对应的类别的,所以需要有 val_map.txt 文件,需要的参考我的blog:制作 ImageNet 验证集的标签 val_map.txt(1000类)
完整代码
from pytorch_grad_cam import GradCAM
可视化工程下载链接
https://download.csdn.net/download/weixin_42899627/76473610
可视化是需要图像和它对应的类别的,所以需要有 val_map.txt 文件,需要的参考我的blog:制作 ImageNet 验证集的标签 val_map.txt(1000类)
from pytorch_grad_cam import GradCAM