格拉德卡姆和格拉德卡姆++的一个简单的蟒蛇实现
pytorch-gradcam的Python项目详细描述
gradcam[1]和gradcam++[2]
的简单pytorch实现安装
pip install pytorch-gradcam
支持的torchvision模型
- 亚历克斯内特
- VGG
- resnet
- 登塞内特
- 挤压网
使用量
如果您想知道如何正确设置target_layer_name，请参考example.ipynb了解一般用法，并参考utils.py中每个层查找函数的文档。
参考文献：
[1]Grad Cam:通过基于梯度的定位从深层网络进行视觉解释，Selvaraju等人，ICcv，2017
[2]grad cam++：基于广义梯度的深卷积网络视觉解释，chattopadhyay等，wacv，2018
欢迎加入QQ群-->： 979659372
