Skip to content

92% accuracy CNN from scratch for CIFAR-10 dataset WITHOUT transfer learning and with complex metrics.

Notifications You must be signed in to change notification settings

Delshi/cifar10_cnn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

The cifar-10 CNN classifier with ~92% accuracy. It uses some augmentation of images, contains logging of lots of metrics such as projector, PR-curves, etc.

How to use

Installation

  1. Run requirements.bat and then start model.py
  2. Unpack archives in logs folder

Usage

  1. To start learning, run start_tb.bat and then run model.py
  2. Before learning delete logs directory, then run model.py
  3. To see model predictions after learning, run model.py
  4. DATA directory contains results of my attempt of learning
  5. Logs directory contains all the results of learning

Results

Graphs

Epoch Acc and Loss
Acc_Loss Plot
Epoch LR
Epoch Learning Rate

Confusion Matrix

Epoch 3
Confusion Matrix epoch 3
Epoch 84
Confusion Matrix epoch 84

PR Curves

Automobile Epoch 1
PR Curve automobile epoch 1
Automobile Epoch 84
PR Curve automobile epoch 84
Cat Epoch 1
PR Curve cat epoch 1
Cat Epoch 84
PR Curve cat epoch 84

Projector

PCA
PCA
UMAP 2D
UMAP 2D
UMAP 3D
UMAP 3D

Histograms

Gradients Beta
Gradients Beta
Gradients Bias
Gradients Bias
Main Kernel
Kernel
Moving Mean
Moving Mean

Distributions

Gradients Gamma
Gradients Gamma
Gradients Kernel
Gradients Kernel
Main Kernel
Kernel
Moving Mean
Moving Mean
Moving Variance
Moving Variance

More in a DATA directory