| --- |
| 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 |
