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

最近写论文需要画热力图,感觉不少论文里都是用Grad-CAM来画,就尝试了一下。
具体步骤:
1. 进入gitthub下载压缩包文件:https://github.com/jacobgil/pytorch-grad-cam
Grad-CAM绘画热力图 使用教程 pycharm+anaconda 论文画图 卷积神经网络CNN Resnet_第1张图片

也可以点击链接 CSDN积分下载
2. 解压后在pycharm中打开文件
Grad-CAM绘画热力图 使用教程 pycharm+anaconda 论文画图 卷积神经网络CNN Resnet_第2张图片
3. 安装缺少的pytthon包,打开anaconda prompt, activate环境,pip install +包名 下载(conda install也行)
4. 运行 目录中的cam.py文件。压缩包里带了例图,运行时会下载Resnet的模型,结果图保存在cam.py文件相同目录下。
Grad-CAM绘画热力图 使用教程 pycharm+anaconda 论文画图 卷积神经网络CNN Resnet_第3张图片5. 运行后会保存三张图

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

Grad-CAM绘画热力图 使用教程 pycharm+anaconda 论文画图 卷积神经网络CNN Resnet_第5张图片
Grad-CAM绘画热力图 使用教程 pycharm+anaconda 论文画图 卷积神经网络CNN Resnet_第6张图片
代码中注释很详细,就不具体说了,有问题可以评论区留言
参考文章: https://www.jianshu.com/p/1abb721b3c99
grad-CAM论文地址:https://link.springer.com/article/10.1007/s11263-019-01228-7

你可能感兴趣的:(pycharm,cnn,深度学习,图像识别,图像处理)