Simpsons-CNN

Overview

• CNN model to classify characters from The Simpsons
• Trained using TensorFlow 2.0
• Run inference on an Android device

Data

• The dataset is composed of images of characters from The Simpsons provided by alexattia

Methodology

• Using ImageDataGenerator to augment training data set
• Using MobileNetV2 as the base model
• Using TensorFlow Lite Converter to convert trained model for use in an Android app

Results

Image	Prediction

Training after 15 epochs

Training after 15 epochs + 15 epochs of fine-tuning