Data science project example using deep learning for image classification

Overview

For this example project I built a ball classifier to identify balls from different sports. This could be useful for someone who is new to sports from a certain country. They could take a picture of a ball and an app could serve them some information about the history and rules of the game. This is the underlying model for building something with those capabilities.

I was able to get the model to predict the sport of the ball with 94% accuracy after minimal tuning. For most of the cases this would meet the need of an end user of the app. To get these results I used transfer learning on a CNN trained on resnet34. This created time efficiencies and solid results.

Notes

This notebook takes you through the process of creating an image classifer for various types of sports balls. It is based of the fastai lesson two notebook.

Data

I used the following chrome extension to download the data from google images. FatKun Batch Download Image