vb690 / cnn_representation_visualizer

Small app for visualizing changes in filters and representation produced by a Convolutional Neural Network during training.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Anatomy of a Convolutional Neural Network

License: MIT Open in Streamlit

Motivation

This App aims to visualize changes in filters and representation produced by a Convolutional Artificial Neural Network (LeNet-5) while learning how to classify different clothing categories present in the Fashion-MNIST dataset.

For maximizing the observed differences and reducing memory overhead, the observed training period is represented by the sequence of random batches observed during a single training epoch.

Features

  • Script for trainining-on-batch of LeNet-5 on Fashion-MNIST.
    • Extracting learned filters over training.
    • Extracting learned embeddings oser training.
    • Running AlignedUMAP ovser sequence of extracted embeddings.
  • Streamlit App
    • Visulize changes in learned filters over training.
    • Visulize changes in UMAP reduction of learned embedding over training.

How to use

Open the app clicking on the streamlit badge on top of this README.

License

The MIT License

About

Small app for visualizing changes in filters and representation produced by a Convolutional Neural Network during training.


Languages

Language:Python 100.0%