/legos

Classification of different Lego mini figures

Primary LanguageJupyter Notebook

legos

Classification of different Lego mini figures using keras and tensorflow in Python.

Data can be found here: https://www.kaggle.com/ihelon/lego-minifigures-classification

The project aims at classifying images as different classes of legos using a pre-trained model in keras, VGG16. The notebook is organised as follows:

  1. Data exploration
  2. Train and validation split
  3. Training image generation in batches using Data generator
  4. Visualization
  5. Training a VGG16 model
  6. Evaluating model accuracy and loss