Skip to main content

Generate a neural network architecture Image

Project description

Neural Network Visualizer

General Description

A module which creates a neural network image with the given architecture. It's a handy tool to see how your network is built as compared to a model summary.


Before installation

Before installing the module, run the below command at your prompt to install the graphviz

$ sudo apt install graphviz

Normal installation

$ sudo pip3 install neuralnet-visualize

Development installation

$ git clone
$ cd nn_visualize

After installation

After installing the module, if you want to upgrade the module, run the below command.

sudo pip3 install neuralnet-visualize --upgrade

Future Works

  • <input type="checkbox" checked="" disabled="" /> Add Convolutional layers, Maxpooling, Flatten layers
  • <input type="checkbox" disabled="" /> Add Sequence model layers
  • <input type="checkbox" disabled="" /> Directly from the pickle files
  • <input type="checkbox" disabled="" /> Specific colors for activation functions
  • <input type="checkbox" disabled="" /> Directly convert from pytorch models

Neural Network Visualizer Version History


  • Added from_pytorch method
  • Fixed a bug of multiple titles in an image
  • Changed the logic of from_tensorflow method
  • Fixed a bug, unique layer names for non trainable layers


  • Added more docstring
  • Can now be downloaded in py2


  • Added title to the image
  • Optimized some code


  • Made it independent of tensorflow
  • Fixed some bugs


  • Fixed some bugs
  • Added more layers maxpooling, avgpooling, flatten
  • changed the shape of conv2d layer


  • Added class docstrings with examples
  • Added some more parameters for conv layer
  • Disabled custom addition of layers after from_tensorflow call


  • Added Conv layer, with kernel size parameter
  • Refactored add_layer function
  • Added Conv layer to from_tensorflow


  • Added some possible exceptions
  • Restructred basic functions in while initial import


  • Added documentation


  • Refactored summarize function
  • Added a code part which colors output layer to red
  • Added from_tensorflow method


  • Added more display orientations
  • Refactored the build network and add layer code
  • Added some more file extensions to which image can be saved
  • Added summarize function


  • Added colors to type of layers
  • Added Dense layers


Intial Version

  • Starting code

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for neuralnet-visualize, version 0.2.3
Filename, size File type Python version Upload date Hashes
Filename, size neuralnet_visualize-0.2.3-py3-none-any.whl (12.1 kB) File type Wheel Python version py3 Upload date Hashes View
Filename, size neuralnet_visualize-0.2.3.tar.gz (7.7 kB) File type Source Python version None Upload date Hashes View

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page