Giters

paulgavrikov / visualkeras

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.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[REQUEST] Add layer name and tensor shapes

ekardnam opened this issue · comments

commented

This is very quick to deploy to Google Colab and I dont have to install everything on my computer which is great, but it would be nice if there was the possibility to have the layer names as well as the tensor shapes plotted (i.e. for convolutional layers height x width @ kernel number).

The only alternative to this is opening the file on photoshop and editing it by hand which isnt very good for productivity i guess.

commented

The addition of this function would make the application perfect

commented

I added the tensor shapes as text below the layers in my fork and just created a pull request so that this feature finds its way into the main repository.

commented

Stale issue message