/fastai-waste-classifier

Primary LanguageJupyter Notebook

Using FastAI for creating a Waste Classifier

Fastai is a deep learning library which provides high-level components that can quickly and easily provide state-of-the-art results in standard deep learning domains.

It has two main design goals:

To be approachable and rapidly productive To be also configurable.

Waste Classifier

The aim is to build a model for waste classification that identifies among the different classes:

  • cardboard
  • compost
  • glass
  • metal
  • paper
  • plastic
  • trash

This machine learning model will help people to improve their decision when classifying trash

Dataset

The data is already splitted in train and test folders. Inside each folder contains one folder for each class. Those images were obtained using Bing searcher using the api HTTP. You can find the code used to download the images at this Google Colab. You can find the final dataset splitted into train and test here in a Google Drive folder.

Split into train/test

Model

You can find the code for the classifier resnet-model.ipynb To load the model in the file result-resnet34.pkl

The model reaches an accuracy of 0.98, here we can see the confusion matrix: confusion_matrix

Build image

docker buildx build --platform linux/amd64 -t waste-classifier .

Web

The model is uploaded at this url http://waste-classifer.rs-sandbox.net/

Google Colab - use the model

Check this Google Colab to use the model with your own images.