Project information

  • Category: Artificial Intelligence
  • Client: SUTD (school)
  • Project date: 01 - 04/2024 (Term 8)

CanOrNot

Our task was to use ML models to solve a problem. Our motivation was the tons of waste and low recycling rate.

Our objective was to develop a highly accurate ML model that can correctly classify waste materials into their respective categories, and thus promote sustainability and recycling practices. We created a recycling web application that takes in images of items, detects and classifies them according to the material and category through our machine learning model.

Tech Stack

  • ReactJS
  • FastAPI
  • Google Colab

My Main Contributions

  • Problem and solution suggestions
  • VGG-16 model
  • Evaluation of all models and comparison using metrics like accuracy, F1 score, confusion matrix
  • More Details