raghakot / keras-vis

Neural network visualization toolkit for keras
https://raghakot.github.io/keras-vis
MIT License
2.97k stars 664 forks source link

Add keepdims option to the visualize_saliency* functions #137

Closed keisen closed 5 years ago

keisen commented 5 years ago

This PR add options to output colored saliency map (i.e., saliency map containing channels). As side effects, it’s able to avoid the problem of non-image data (i.e., no channel data) like issues / #135 .

@raghakot, I made this PR as a change log. So I’m going to self-merge, but please revert if there is any problem.

raghakot commented 5 years ago

LGTM thanks.