Python Engineer

Python and Machine Learning Tutorials

back to course overview

Classify Lego Star Wars Minifigures - TensorFlow Beginner 08

09 Oct 2020

In this part we use a real image dataset from Kaggle with Lego Star Wars Minifigures and do a full project walkthrough. I show you how to load the data using a tensorflow ImagaDataGenerator and implement a convolutional neural net to do image classification. I also introduce some new concepts like image augmentation and keras callbacks.

All code and slides from this course can be found on GitHub.

Download the dataset: