README.md

metadata

title: Trashify Demo V2 🚮
emoji: 🗑️
colorFrom: purple
colorTo: blue
sdk: gradio
sdk_version: 4.40.0
app_file: app.py
pinned: false
license: apache-2.0

🚮 Trashify Object Detector Demo V2

Object detection demo to detect trash, bin, hand, trash_arm, not_trash, not_bin, not_hand.

Used as example for encouraging people to cleanup their local area.

If trash, hand, bin all detected = +1 point.

Dataset

All Trashify models are trained on a custom hand-labelled dataset of people picking up trash and placing it in a bin.

The dataset can be found on Hugging Face as mrdbourke/trashify_manual_labelled_images.

Demos

• V1 = Fine-tuned DETR model trained without data augmentation.
• V2 = Fine-tuned DETR model trained with data augmentation.
• V3 = Fine-tuned DETR model trained with data augmentation (same as V2) with an NMS (Non Maximum Suppression) post-processing step.

TK - finish the README.md + update with links to materials