Skip to content

Aishwar1/CNN-Visualizer

Repository files navigation

🧠 CNN Feature Visualization Tool

A deep learning visualization system that analyzes and visualizes how Convolutional Neural Networks (CNNs) extract features across layers.
This project focuses on understanding what CNNs learn internally by visualizing feature maps, weights, and layer-wise behavior across popular architectures.


🚀 Project Overview

This tool explores how CNN architectures progressively learn edges, textures, shapes, and spatial hierarchies from images.
Instead of treating CNNs as black boxes, the project exposes internal representations to improve interpretability and debugging of deep learning models.


🧩 Supported Models

  • VGG16
  • VGG19
  • ResNet
  • Inception

🔄 System Flow (Clear & Visual)

flowchart TD
    A[Input Image] --> B[Preprocessing\nResize, Normalize]
    B --> C[CNN Model\nVGG16 / VGG19 / ResNet / Inception]
    C --> D[Convolution Layers]
    D --> E[Feature Maps\n64 Channels]
    E --> F[Visualization\nEdges, Textures, Patterns]
    F --> G[Layer-wise Analysis\nWeights & Biases]
    G --> H[Metrics Evaluation\nLoss, MSE, R²]
Loading

✨ Key Features

  • Visualization of 64-channel feature maps from convolutional layers
  • Analysis of edges, textures, and spatial patterns learned by CNNs
  • Layer-wise inspection of weights, biases, and activations
  • Comparison across multiple CNN architectures
  • TensorBoard integration for training and feature tracking
  • Evaluation using Loss, Mean Squared Error (MSE), and R² metrics

🛠️ Tech Stack

Core Technologies

  • Python
  • TensorFlow / Keras
  • NumPy
  • Matplotlib

Deep Learning

  • Convolutional Neural Networks (CNNs)
  • Transfer Learning (VGG, ResNet, Inception)

Visualization & Analysis

  • Feature map visualization
  • TensorBoard
  • Layer-wise interpretability

📂 Project Structure

cnn-feature-visualization/
│
├── models/
│   ├── vgg16
│   ├── vgg19
│   ├── resnet50
│   └── inceptionV3
│
├── static/
|   └── script.js
│
├── templates/
│   └── index.html
├── app.py
│
├── model_utils.py
└── README.md

▶️ How to Run

python -m venv venv
source venv/bin/activate
pip install -r requirements.txt
python app.py

🎯 Project Goals

  • Improve interpretability of CNN-based models
  • Provide visual intuition behind deep learning decisions
  • Help debug and compare CNN architectures
  • Bridge theory and practical understanding of convolutional networks

About

CNN Feature Visualization Tool: A deep learning visualization framework built with TensorFlow to analyze and interpret convolutional neural networks by visualizing feature maps, layer-wise activations, and learned representations across VGG, ResNet, and Inception architectures.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors