We were inspired by the recent climate change march and we wanted to create an environment friendly application.

What it does

Basically, we built a mobile application which allows the user to efficiently determine the category of waste he is trying to dispose by taking a picture of it.

How we built it

We trained a convolutional neural network to classify an image as either cardboard, glass, metal, paper, plastic, or trash with the fastai library (built on PyTorch). The pre-trained model is then used to predict the classification of images received through an Android application built on ReactNative and Flask.

Challenges we ran into

Due to the duration of the event, it was difficult to train the model for multiple iterations in order to increase its accuracy, which we managed to do, although not perfectly, overnight.

What we learned

Since we are relatively new to the field of machine learning, we are satisfied to have furthered our understanding about convolutional neural networks and image classification

What's next for GoGreen

Since the duration was relatively short, in a future project, more thorough data scraping should be used in order to implement more categories of waste such as compostable elements and electronics.

Built With

Share this project: