keras-vis icon indicating copy to clipboard operation
keras-vis copied to clipboard

Support _Conv.data_format for visualize methods

Open Qinusty opened this issue 5 years ago • 0 comments

Currently visualize_cam and visualize_saliency assume models follow the standard keras.backend.image_data_format() which leaves the function unusable with models which aren't using this.

See saliency.py L174 for an example of usage here.

Potentially adding a data_format parameter or extracting this information from the penultimate_layer with penultimate_layer.data_format .

Qinusty avatar Apr 18 '19 17:04 Qinusty