Generating a heatmap from a image

So I've have trained a simple image classifier in TensorFlow 2.0, I based it on this, it uses the second method from the website because the first one, where it doesn't use Dropout or Data augmentation overfits a bit.

So now I need to generate a heatmap for where it thinks the class is (I don't need it for flowers like the TensorFlow tutorial).

I watched a Computerphile video which suggests that I should use a CNN for this problem.

So if I do train a network like this would it need photos with the class's positions or just images?

Please answer.

1 answer