0
点赞
收藏
分享

微信扫一扫

Grad-CAM绘画热力图 使用教程 pycharm+anaconda 论文画图 卷积神经网络CNN Resnet

最近写论文需要画热力图,感觉不少论文里都是用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

举报

相关推荐

0 条评论