raghakot / keras-vis

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

gif generator callback error #155

Closed ju-w closed 5 years ago

ju-w commented 5 years ago

This will throw an error

https://github.com/raghakot/keras-vis/blob/668b0e11dab93f3487f23c17e07f40554a8939e9/vis/callbacks.py#L60-L63

because deprocess_input's output must be converted to be used in gif

A fix would be:

         img = utils.deprocess_input(wrt_value[0]).astype('uint8')

Demo Generating animated gif of optimization progress from Documentation will currently not work.