Skip to content

jimgitonga/Multiclass-Image-Classifier-pytorch-Transfer-Learning

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 

Repository files navigation

Multiclass-Image-Classifier-pytorch-Transfer-Learning

This is a multi-class image classifier that have 8 classes and only few images in the training set for each class.

The dataset used here is a derivation of the “Natural Images” dataset (https://www.kaggle.com/prasunroy/natural-images/version/1#_=_ ) . The number of images in the original dataset has been reduced randomly for building the “Mini Natural Images”. This dataset consists of three phases for train, test and validation. (The dataset is available in the GitHub repository) Here’s an overview of the “Mini Natural Images” dataset.

A complete desciption on the experiment is available here. ( https://naadispeaks.wordpress.com/2018/11/24/transfer-learning-in-convnets-part-2/ )

About

A multi-class image classifier using transfer learning with pytorch.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%