README.md

metadata

title: Traffic Sign Detection
emoji: 🔥
colorFrom: indigo
colorTo: gray
sdk: gradio
sdk_version: 5.42.0
app_file: app.py
pinned: false

Traffic Sign Detection

This project features a YOLOv8-based model for detecting and classifying traffic signs in images, integrated with a Gradio interface for interactive use. Upload an image to view detected traffic signs with bounding boxes and labels, powered by a model trained on the German Traffic Sign Recognition Benchmark (GTSRB) dataset.

Usage

1. Upload an Image: Use the Gradio interface to upload an image containing traffic signs.
2. View Results: The app displays the image with bounding boxes and labels for detected traffic signs.
3. Confidence Threshold: Detections are filtered at a confidence score of 0.5 (adjustable in app.py if needed).

Model Details

• Architecture: YOLOv8 nano
• Dataset: GTSRB, 43 classes, ~39,209 training images
• Training: 50 epochs, 640x640 image size, 80% training / 20% validation split

Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference