从图像生成热图

Generating a heatmap from a image

所以我在 TensorFlow 2.0 中训练了一个简单的图像 classifier,我基于 this,它使用网站上的第二种方法,因为第一种方法,它没有不要使用 Dropout 或数据扩充过度拟合。

所以现在我需要为它认为 class 所在的位置生成一个热图(我不需要像 TensorFlow 教程那样的花)。

我看了一个 Computerphile video,它建议我应该使用 CNN 来解决这个问题。

因此,如果我像这样训练网络,它需要带有 class 位置的照片还是只需要图像?

请回答。

您需要类似 Grad-CAM ”的技术。一种通过可视化对预测“重要”的输入区域,使基于卷积神经网络 (CNN) 的模型更加透明的技术来自这些模型 - 或视觉解释。"

这里有一些链接从一开始就涵盖了它。

https://www.pyimagesearch.com/2020/03/09/grad-cam-visualize-class-activation-maps-with-keras-tensorflow-and-deep-learning/

https://medium.com/@mohamedchetoui/grad-cam-gradient-weighted-class-activation-mapping-ffd72742243a

以下为论文原文:

http://gradcam.cloudcv.org/