最近写论文需要画热力图,感觉不少论文里都是用Grad-CAM来画,就尝试了一下。
具体步骤:
1. 进入gitthub下载压缩包文件:https://github.com/jacobgil/pytorch-grad-cam
也可以点击链接 CSDN积分下载
2. 解压后在pycharm中打开文件
3. 安装缺少的pytthon包,打开anaconda prompt, activate环境,pip install +包名 下载(conda install也行)
4. 运行 目录中的cam.py文件。压缩包里带了例图,运行时会下载Resnet的模型,结果图保存在cam.py文件相同目录下。
5. 运行后会保存三张图
代码中注释很详细,就不具体说了,有问题可以评论区留言
参考文章: https://www.jianshu.com/p/1abb721b3c99
grad-CAM论文地址:https://link.springer.com/article/10.1007/s11263-019-01228-7