Skip to content

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow

Notifications You must be signed in to change notification settings

himanshugupta1009/OscarNET

 
 

Repository files navigation

OscarNET: Garbage Detection and Extraction

The world is currently in the grips of a trash crisis. It is estimated that the a person produces approximately 234lb (106.2kg) of plastic waste per annually. The enormous amount of waste produced does not get discarded properly and ends up on the roadside, or in parks and wildlife areas. Our goal is to address this issue by developing an autonomous system that classifies and then removes trash from public spaces. We implemented a Mask R-CNN based model to detect garbage in images and create segments over it. We use the transfer learning approach to reduce training time by using a model trained on the famous COCO dataset. In conjunction, we use a pre-existing image depth estimator to estimate the depth and aid autonomous navigation. The current implementation sends Sparki bot to extract the garbage but this can be extended on any bot.

Our model OscarNet, named after the garbage loving character from the show Sesame Street, was trained on the dataset that we curated.

Figure 1: Detecting garbage in a park OscarNET at work

Figure 2: Team meber Hunter posing with garbage for OscarNET Team member Hunter posing with garbage for OscarNET

Figure 3: Garbage detection during our experimental setup for dispatching Sparki bot and extracting garbage Team member Hunter posing with garbage for OscarNET

Figure 4: Detecting garbage bags on street Team member Hunter posing with garbage for OscarNET

About

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow

Resources

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 54.8%
  • Jupyter Notebook 43.9%
  • Shell 1.3%