/ecosnap

Recycle your plastic better with Artificial Intelligence ♻️

Primary LanguagePureBasic

EcoSnap

ecosnap.mp4

Recycle your plastic better with Artificial Intelligence ♻️

EcoSnap tells you what, how, and where to recycle your items from a simple picture using Artificial Intelligence. You just take a picture of a plastic resin code, and it tells you how and where to recycle the item based on your location. We built this product in a week for Ben's Bites Hackathon.

👉 Try it now - it's free with no sign in needed

Deploy with Vercel

You can support this project (and many others) through GitHub Sponsors! ❤️

Made by Alyssa X & Leo

Table of contents

Features

📸 Snap a picture of a plastic code
🔍 Search for specific item
♻️ Learn how to recycle effectively using AI
🥤 Keep track of how many plastic items you've recycled
🌍 Change your location for specific advice
✨...and much more to come - all for free & no sign in needed!

The AI Model

Data

The model was trained on image examples of the 7 different resin codes, the data for this can be found in ml/seven_plastics. It is a combination of the following Kaggle Dataset and images collected by the authors and contributors.

Training

The final model was trained using TensorFlow's EfficientNet implementation, the model weights were frozen for transfer learning, so the model could learn the resin codes faster! The model was trained in Python on a GPU-powered machine, for faster training! You can find the training script in ml/train.py and try it for yourself, there you will see that different meta architectures and parameters were experimented with before arriving at the final model.

Prediction

To predict the plastic resin code, the model had to be integrated with the front end app for real-time results, to do this we had to convert the model in a way that was compatible with Node.js, this is when TensorFlow.js came to the rescue.

The app passes the image Tensor onto the model that then gives a probability for each of the plastic resin codes, the one with the highest probability gets shown to the user, along with bespoke advice!

Feedback

Training a specific model is hard, the model always gets things wrong. So if it does, we give the user an opportunity to tell us what the right code was! This benefits in several ways:

  1. The user gets the information they need on how to recycle their item
  2. We can see how the model is performing in production
  3. We get new data ( if the user lets us) to train the model with and improve it for everyone

Credit

Libraries used

What's next?

We want to implement a few more ideas after the Hackathon:

  1. A map that shows you where your nearest recycling point is
  2. Top tips on how to recycle
  3. Articles and research about plastic
  4. Get the user's permission to use their images for retraining the model

Feel free to reach out to us at hi@alyssax.com, to Alyssa or Leo directly if you have any questions or feedback! Hope you find this useful 💜