trashify_demo_v1 / README.md
mrdbourke's picture
Uploading Trashify box detection model app.py
27b9ba6 verified

A newer version of the Gradio SDK is available: 5.12.0

Upgrade
metadata
title: Trashify Demo V1 ๐Ÿšฎ
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 V1

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 - add links to resources to learn more