Skip to content

Aftaab99/PyTorchImageClassifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PyTorch image classifier for the notMNIST dataset

The notMNIST dataset contains 28x28px images of letters A to J in different fonts. This repository aims to serve as a sample for image classification in pytorch. It includes

  1. Custom Dataset and dataloaders
  2. Training and testing the model
  3. Creating a multiclass image classification model

Dataset

A B C D E F G H I J

The dataset contains over 18500 images for training and 469 images for testing. All images are grayscale 28x28px images.

Model used and accuracy

A convolutional neural network achieves an accuracy of 92.37% on this dataset.

About

A pytorch image classifier for the recognising letters from the notMNIST dataset

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages