Plastic Detection AI

Inspiration

Our inspiration for this project is the current global issue of plastic pollution.

What We Learned

Throughout this project, we delved into the capabilities of deep learning to solve real-world problems. We worked with the YOLO (You Only Look Once) object detection model and adapted it to the specific task of detecting plastic waste.

How We Built It

Our project is based on Python and utilizes the Ultralytics implementation of YOLO. We started by annotating a dataset of aerial images containing visible waste. The images were labeled with bounding boxes around each piece.

We then trained our YOLO model on this dataset, fine-tuning the weights to accurately detect plastic waste.

Built With

Share this project:

Updates