collindching / waste-sorter

An image classifier for disposables, written in Python

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Waste-Sorter

This was my first attempt at building an image classifier using the fastai library, trained to sort different types of trash. The test accuracy was really high, 92%. I also wrote a code walkthrough on Medium.

All the code for this project is in Waste sorter.ipynb.

How to view this project

The notebook is too big to render on Github, which is why we have to use Jupyter Notebook Viewer to render it. Start with option (a) and if the link doesn't work, use option (b).

Option (a)

Option (b)

  1. Copy the link to the notebook on Github (https://github.com/collindching/Waste-Sorter/blob/master/Waste%20sorter.ipynb)
  2. Follow this link here and paste the notebook URL to view the notebook

How to run this notebook

  1. Download Waste sorter.ipynb
  2. Download the corresponding dataset here and move it to the same directory as the notebook
  3. Run the notebook in Anaconda

About

An image classifier for disposables, written in Python


Languages

Language:Jupyter Notebook 100.0%