ebeyabraham / tsne-viz

t-SNE visualisation of CNN features

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

t-SNE Visualisation of CNN Features

t-SNE visualisation of CNN features can help understand how the CNN understands the relationship between different classes which it is trained on. This repository contains minimal code to visualise t-SNE embeddings of the features obtained after average pooling in the ResNet-18 architecture. For visualisation the validation set of Imagenette was used.

Usage

  1. The jupyter notebook generates .tsv files containing the features (vis.tsv) and the corresponding labels (metadata.csv)
  2. Upload both files to : https://projector.tensorflow.org/
  3. Et voila!

projector

About

t-SNE visualisation of CNN features

License:MIT License


Languages

Language:Jupyter Notebook 100.0%