Sawon1234 / Netplot

A ultra-lightweight 3D renderer of the Tensorflow/Keras neural network architectures

Netplot🚀

A ultra-lightweight 3D renderer of the Tensorflow/Keras neural network architectures. This Library is working on Matplotlib visualization for now. In future the visualization can be moved to plotly for a more interactive visual of the neural network architecture.

Note: For now the rendering is working in Jupyter only Google Colab support is in works.

For more details visit NetPlot

How to use it

Install with Pip

pip install netplot

Notebook Codelets

from netplot import ModelPlot
import tensorflow as tf
import numpy as np

%matplotlib notebook

X_input = tf.keras.layers.Input(shape=(32,32,3))
X = tf.keras.layers.Conv2D(4, 3, activation='relu')(X_input)
X = tf.keras.layers.MaxPool2D(2,2)(X)
X = tf.keras.layers.Conv2D(16, 3, activation='relu')(X)
X = tf.keras.layers.MaxPool2D(2,2)(X)
X = tf.keras.layers.Conv2D(8, 3, activation='relu')(X)
X = tf.keras.layers.MaxPool2D(2,2)(X)
X = tf.keras.layers.Flatten()(X)
X = tf.keras.layers.Dense(10, activation='relu')(X)
X = tf.keras.layers.Dense(2, activation='softmax')(X)

model = tf.keras.models.Model(inputs=X_input, outputs=X)

modelplot = ModelPlot(model=model, grid=True, connection=True, linewidth=0.1)
modelplot.show()