Demonstrate layer dimension in ‘visualcross’

As a deep learning specialist or student, you must work with documentation. Your document must show your models in detail. When you search Google about “how to demonstrate our visulate deep learning model details”, you’ll find ‘visualcross’ library.

Paul Gavrikov developed ‘visualcross‘.

whats visualcross?

Visualkeras is a Python package to help visualize Keras (either standalone or included in tensorflow) neural network architectures. It allows easy styling to fit most needs. This module supports layered style architecture generation which is great for CNNs (Convolutional Neural Networks), and a graph style architecture, which works great for most models including plain feed-forward networks.

 

Demonstrate layer dimension in ‘visualcross’

If you installed this library you probably asked “how to show layer dimensions detail?”. In order to solve this issue, I forked this repository and added new lines inside the source code for this library.

While I submitted a pull request to merge my contribution into the main code, the repository owner has not approved it yet.

If you want to use my code, you can download, install, and use it manually.

If you need to show layer dimension you must set legend=True and show_dimantion=True in layered_view:

visualkeras.layered_view(model, legend=True, show_dimantion=True)