Python实现的Grad-CAM for Pytorch

Grad-CAM的Pytorch实现。Pytorch implementation of Grad-CAM Pytorch 中的 Grad-CAM 实现是什么让网络认为图像标签是“哈巴狗，哈巴狗”和“虎斑猫，虎斑猫”：将 Grad-CAM 与引导反向传播结合用于“哈巴狗，哈巴狗”类：梯度类激活图是深度学习网络的一种可视化技术。 见论文：https://arxiv.org/pdf/1610.02391v1.pdf 论文作者火炬实现：https://github.com/ramprs/grad-cam 我的 Keras 实现：https://github.com/jacobgil /keras-grad-cam 这使用来自 torchvision 的 VGG19。 第一次使用时会下载。 可以修改代码以适用于任何模型。 然而，torchvision 中的 VGG 模型具有用于网络卷积部分和全连接部分的特征/分类器方法。 此代码假定传递的模型支持这两种方法。 用法：python gradcam.py --image-path 与 CUDA 一起使用： python gradcam