keras-grad-cam icon indicating copy to clipboard operation
keras-grad-cam copied to clipboard

Loss function

Open vianamp opened this issue 7 years ago • 2 comments

Hi. Good job putting this together. I was wondering whether loss=K.sum(...) should be replaced with K.max(...) once we are interested on the output for the class with highest output value only. What do you think?

vianamp avatar Oct 31 '17 15:10 vianamp

Notice that the last layer type is target_category_loss,
K.sum here is the same as K.max, only a single class is activated.

jacobgil avatar Nov 03 '17 10:11 jacobgil

Absolutely. Thanks.

vianamp avatar Nov 03 '17 12:11 vianamp