- Original Impl: https://github.com/jacobgil/pytorch-grad-cam
- Grad-CAM简介: https://b23.tv/1kccjmb
- 使用Pytorch实现Grad-CAM并绘制热力图: https://b23.tv/n1e60vN
- 将创建模型部分代码替换成自己创建模型的代码,并载入自己训练好的权重
- 根据自己网络设置合适的
target_layers
- 根据自己的网络设置合适的预处理方法
- 将要预测的图片路径赋值给
img_path
- 将感兴趣的类别id赋值给
target_category