About feature visualizing

Hi,

Thanks for the excellent code. I successfully trained a model with my own customized dataset. And now I want to visualize what is learned by the neural network. Do u have any ideas on doing that?

I tried Lucid, but it seems that I need to modify the input parts. I am wondering whether this is correct?