I've been playing around with Keras's built-in visualizations tool for a while and I feel like it could be improved a lot. The current visualization system doesn't provide much information "at a glance" (meaning it takes quite a bit of analysis to understand what's going on) and isn't super aesthetically pleasing.

I'm opening this issue to see if an improved visualization system is something that the Keras team would be open to. I'd be happy to make small improvements to the existing system, but I feel like it would be better to add a new visualization system (or submodule) for highly customizable visual architecture generation. I could bring some (or most) of the key features from visualkeras, a repository I maintain, into the src/visualization module.

Since we'd isolate the dependencies of the new features from Keras as a whole (like what was done for the last system), we wouldn't be adding any bloat through more dependencies.

Below are a few examples of what the visualizations I'd add may look like

Image

Image

Image

Image

Image

Comment From: mattdangerw

I think we'd be most interested in visualizations that could work for generic functional models (not purely sequential models with a one input and one output constraint). Even for convnets you'll usually have residuals, so most real world model will not be purely sequential.

What would the "graph view" you are proposing look like, and how would it compare to plot_model?