trashify_demo_v2 / README.md
mrdbourke's picture
Uploading Trashify V2 box detection model (with data augmentation) app.py
aab9a3a verified

A newer version of the Gradio SDK is available: 5.7.1

Upgrade
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