Return heatmaps instead of color-mapped results

raghakot / keras-vis

Neural network visualization toolkit for keras

https://raghakot.github.io/keras-vis

MIT License

2.98k stars 664 forks source link

Return heatmaps instead of color-mapped results #67

Closed FrankNiemeyer closed 7 years ago

FrankNiemeyer commented 7 years ago

Is there a particular reason why visualize_saliency_with_losses and visualize_cam_with_losses returns a color-mapped result instead of the heatmap/grads?

I would suggest returning the result before mapping to a specific colormap like "jet".

See https://github.com/raghakot/keras-vis/blob/master/vis/visualization/saliency.py#L83 https://github.com/raghakot/keras-vis/blob/master/vis/visualization/saliency.py#L196

raghakot commented 7 years ago

That's a good point. Will change it.

LennartPiro commented 6 years ago

@raghakot could you please make a new release and push it to PyPI? Would be great to have this fix when using pip. Thanks!

akuehlka commented 6 years ago

@raghakot, I'd just like you to know that I spent 2 days of work to find out that release 0.4.1 from github is different from release 0.4.1 from PyPI. Thanks @LennartPiro for pointing that out.