trashify_demo_v4 / README.md
mrdbourke's picture
Uploading Trashify box detection model app.py
0f10c71 verified

A newer version of the Gradio SDK is available: 5.34.0

Upgrade
metadata
title: Trashify Demo V4 ๐Ÿšฎ
emoji: ๐Ÿ—‘๏ธ
colorFrom: purple
colorTo: blue
sdk: gradio
sdk_version: 5.29.0
app_file: app.py
pinned: false
license: apache-2.0

๐Ÿšฎ Trashify Object Detector V4

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 Conditional DETR model trained without data augmentation.
  • V2 = Fine-tuned Conditional DETR model trained with data augmentation.
  • V3 = Fine-tuned Conditional DETR model trained with data augmentation (same as V2) with an NMS (Non Maximum Suppression) post-processing step.
  • V4 = Fine-tuned RT-DETRv2 model trained without data augmentation or NMS post-processing (current best mAP).

TK - add links to resources to learn more