An interactive web app to visualize neural network architectures and their activations.
Built with Python + PyTorch for model building, and a lightweight web front-end for visualization.
Tips: Run local for best performance. Thanks
| Feature | Description |
|---|---|
| Custom Model Support | Define any PyTorch model (MLP, CNN, Transformer, etc.) and visualize it directly. |
| Interactive Play Mode | Animate activations flowing through the network with a single click. |
| Layer & Neuron Mapping | Shows layers, neurons, and weighted connections clearly. |
Load .pth Files |
Import and visualize pretrained models saved in PyTorch. |
| Deploy Anywhere | Works locally or on GitHub Pages for free hosting. |
- Provide
Python Modelcode + load.pthmodel file. - Displays layers, neurons, and weighted connections by use
Buildbutton - Interactive
Runbutton to animate activations
git clone https://github.com/Iro96/NeuralNet-Visualizer.git
cd neural-net-visualizerpython server.py- Just open file
index.htmlin root directory (./) - Experience the tool your own way.
Thanks a lot for the help from @ZhenYu15