poloclub / cnn-explainer
Learning Convolutional Neural Networks with Interactive Visualization.
AI Architecture Analysis
This repository is indexed by RepoMind. By analyzing poloclub/cnn-explainer in our AI interface, you can instantly generate complete architecture diagrams, visualize control flows, and perform automated security audits across the entire codebase.
Our Agentic Context Augmented Generation (Agentic CAG) engine loads full source files into context, avoiding the fragmentation of traditional RAG systems. Ask questions about the architecture, dependencies, or specific features to see it in action.
Embed this Badge
Showcase RepoMind's analysis directly in your repository's README.
Repository Summary (README)
PreviewCNN Explainer
An interactive visualization system designed to help non-experts learn about Convolutional Neural Networks (CNNs)
<a href="https://youtu.be/HnWIHWFbuUQ" target="_blank"><img src="https://i.imgur.com/sCsudVg.png" style="max-width:100%;"></a>
For more information, check out our manuscript:
CNN Explainer: Learning Convolutional Neural Networks with Interactive Visualization. Wang, Zijie J., Robert Turko, Omar Shaikh, Haekyu Park, Nilaksh Das, Fred Hohman, Minsuk Kahng, and Duen Horng Chau. IEEE Transactions on Visualization and Computer Graphics (TVCG), 2020.
Live Demo
For a live demo, visit: http://poloclub.github.io/cnn-explainer/
Running Locally
Clone or download this repository:
git clone git@github.com:poloclub/cnn-explainer.git
# use degit if you don't want to download commit histories
degit poloclub/cnn-explainer
Install the dependencies:
npm install
Then run CNN Explainer:
npm run dev
Navigate to localhost:3000. You should see CNN Explainer running in your broswer :)
To see how we trained the CNN, visit the directory ./tiny-vgg/.
If you want to use CNN Explainer with your own CNN model or image classes, see #8 and #14.
Credits
CNN Explainer was created by <a href="https://zijie.wang/">Jay Wang</a>, <a href="https://www.linkedin.com/in/robert-turko/">Robert Turko</a>, <a href="http://oshaikh.com/">Omar Shaikh</a>, <a href="https://haekyu.com/">Haekyu Park</a>, <a href="http://nilakshdas.com/">Nilaksh Das</a>, <a href="https://fredhohman.com/">Fred Hohman</a>, <a href="http://minsuk.com">Minsuk Kahng</a>, and <a href="https://www.cc.gatech.edu/~dchau/">Polo Chau</a>, which was the result of a research collaboration between Georgia Tech and Oregon State.
We thank Anmol Chhabria, Kaan Sancak, Kantwon Rogers, and the Georgia Tech Visualization Lab for their support and constructive feedback.
Citation
@article{wangCNNExplainerLearning2020,
title = {{{CNN Explainer}}: {{Learning Convolutional Neural Networks}} with {{Interactive Visualization}}},
shorttitle = {{{CNN Explainer}}},
author = {Wang, Zijie J. and Turko, Robert and Shaikh, Omar and Park, Haekyu and Das, Nilaksh and Hohman, Fred and Kahng, Minsuk and Chau, Duen Horng},
journal={IEEE Transactions on Visualization and Computer Graphics (TVCG)},
year={2020},
publisher={IEEE}
}
License
The software is available under the MIT License.
Contact
If you have any questions, feel free to open an issue or contact Jay Wang.