trashify_demo_v3 / README.md
mrdbourke's picture
Uploading Trashify box detection model v3 app.py with NMS post processing
d50f451 verified
|
raw
history blame
1.25 kB
metadata
title: Trashify Demo V3 ๐Ÿšฎ
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 V3

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