在Pytorch中实现Grad-CAM
是什么使网络认为图像标签是“ pug,pug-dog”和“ tabby,虎斑猫”:
将Grad-CAM与引导反向传播相结合的“ pug,pug-dog”类别:
梯度类激活图是用于深度学习网络的可视化技术。
参见论文: :
论文作者的火炬实现: :
我的Keras实现: :
这使用了来自torchvision的Resnet50。 首次使用时将被下载。 可以修改该代码以与任何模型一起使用。
用法: python gradcam.py --image-path
要与CUDA一起使用: python gra
1