最近写论文需要画热力图,感觉不少论文里都是用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. 运行后会保存三张图
5. 运行后会保存三张图


 
 代码中注释很详细,就不具体说了,有问题可以评论区留言
 参考文章: https://www.jianshu.com/p/1abb721b3c99
 grad-CAM论文地址:https://link.springer.com/article/10.1007/s11263-019-01228-7









