/Image_classification

CNN from scratch, data augmentation and Transfer learning for image classification

Primary LanguageJupyter Notebook

Image Classification

In this notebook you will have the possibility to do image classification with a CNN from scratch, data augmentation and Transfer learning.

The goal is to present the three approaches and made them available easily for everyone.

This github repository is associated to the article on medium entitle What is Image Classification? Data augmentation? Transfer Learning? submitted in Towards Data science.

Content

  • Introduction
  • Data
    • Load the data
    • Ploting Learning Curves
    • Metrics
  • Methods
    • CNN from scratch
    • Data augmentation
    • Transfer learning
  • Results
  • Discussion
  • Conclusion