在执行Grad-CAM所遇问题
1)
修改后解决
2)
修改后解决,因为numpy需要在cpu上进行,所有需要加上.cpu()
3)plt.matshow(heatmap)出错
原因是get_heatmap()中的mean_gradients = torch.mean(gradients, dim=[0, 2, 3])出错,尺寸不对。
参考上述公式进行修改,成功!
详细的执行步骤见
Grad-CAM对任意CNN网络进行可视化处理教学,普适+好用!!!
在执行Grad-CAM所遇问题
1)
修改后解决
2)
修改后解决,因为numpy需要在cpu上进行,所有需要加上.cpu()
3)plt.matshow(heatmap)出错
原因是get_heatmap()中的mean_gradients = torch.mean(gradients, dim=[0, 2, 3])出错,尺寸不对。
参考上述公式进行修改,成功!
详细的执行步骤见
Grad-CAM对任意CNN网络进行可视化处理教学,普适+好用!!!