Projects using PyTorch in ThaneProjects using PyTorch in Thane A Neural Network Visualization Tool that demystifies AI!
I built an interactive web application using Flask and PyTorch that doesn't just recognize handwritten digits it shows you exactly how the AI "thinks." When you draw a digit (0-9) on the canvas, the app processes it through a Convolutional Neural Network and generates a real-time visualization of every layer: from edge detection in the first convolutional layer, through pattern recognition, pooling, and feature extraction, all the way to the final classification.
The tech stack includes Python, Flask, PyTorch, and vanilla JavaScript for the frontend. What makes this unique is the educational aspect each layer's activations are visualized using matplotlib, showing the 32 filters in Conv Layer 1, the 64 filters in Conv Layer 2, and the 128-neuron fully connected layer.