Object Detection
ultralytics
yolov8
traffic-sign-detection
traffic-light-detection
computer-vision
ocr
tesseract
Instructions to use poorvika4566/YOLOv8-Traffic-Sign-Light-OCR with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- ultralytics
How to use poorvika4566/YOLOv8-Traffic-Sign-Light-OCR with ultralytics:
from huggingface_hub import hf_hub_download from ultralytics import YOLO # pick the weights file from this repo's "Files and versions" tab weights = hf_hub_download("poorvika4566/YOLOv8-Traffic-Sign-Light-OCR", "<weights>.pt") model = YOLO(weights) source = 'http://images.cocodataset.org/val2017/000000039769.jpg' model.predict(source=source, save=True) - Notebooks
- Google Colab
- Kaggle
YOLOv8 Traffic Sign & Light Detection with OCR
A YOLOv8-based computer vision model for detecting traffic signs and traffic lights from images, video, or live webcam input.
The project also integrates Tesseract OCR to read speed-limit values from detected speed-limit signs.
Note: This is a computer-vision demonstration and is not a certified road-safety system. It should not be used to control a vehicle or make safety-critical decisions.
Model
- Architecture: YOLOv8
- Task: Object Detection
- Model: YOLOv8n
- Input size: 640 ร 640
- Training epochs: 10
- Batch size: 16
- Model weights:
best.pt
Detection Pipeline
The system follows this process:
- Capture frames from a webcam or video.
- Run YOLOv8 object detection.
- Detect traffic lights and traffic signs.
- Identify speed-limit sign detections.
- Crop the detected speed-limit sign.
- Apply image preprocessing.
- Use Tesseract OCR to read the speed value.
- Validate the OCR result against accepted speed-limit values.
- Display the detected sign and recognized speed.
- Save detection results to a CSV file.
OCR
Tesseract OCR is used as a second verification step for speed-limit signs.
The current implementation accepts the following speed values:
5, 10, 15, 20, 25, 30, 60, 80
- Downloads last month
- 82