Trash AI

AI for Litter Detection

Moore Institute for Plastic Pollution Research; Code for Sacramento; CalTrans; Dr. Win Cowger


Trash AI is an open source computer vision artificial intelligence model that allows users to upload images containing trash and returns data about the trash in the image, including the classification of trash and bounding box of where the trash is in the image. Trash AI builds a model using the YoloV5 toolset trained on the TACO dataset. The model takes an image containing trash and returns a list of annotations and bounding boxes of trash within the image. The model is imported into the front-end Vue.js application where it is invoked when an image is uploaded. The Vue application then displays the results of the model on the image.