Aerolite

This machine learning model can distinguish between forks, knives, and spoons. I built it because I wanted to see if I could create my own machine learning model from scratch.

How it works?

The program uses Deep Learning to classify images of cutlery. It uses the TensorFlow library to do so. If you want to learn more about Deep Learning check out this five minute video.

Deatils for nerds

I used TensorFlow Keras to build and train a CNN model. It uses EfficientNet B0 as its backbone with fine-tuning. Half of the images are taken by myself, the other half are from these Kaggle datasets (Knife dataset, Knife vs Pistol, Spoon vs Fork) and web scraping.

View the notebook for the project here.