Skip to content

Latest commit

Β 

History

2 Commits

Folders and files

NameName
Last commit message
Last commit date
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

🚧 Rumble Strip Detection & Driver Warning System

Overview

This project uses Computer Vision and Object Detection to identify highway rumble strips from forward-facing road video and provide an early warning to drivers.

The system was developed using Roboflow RF-DETR and tested on previously unseen highway footage. The objective is to detect rumble strips before the vehicle reaches them, allowing additional reaction time.


Problem Statement

Rumble strips are safety features installed on highways to alert drivers when they are approaching hazardous areas such as:

  • Speed reduction zones
  • Toll plazas
  • Sharp curves
  • Intersections
  • Pedestrian crossings

However, drivers may sometimes fail to notice them until they are very close.

This project aims to automatically detect rumble strips from video feeds and generate an early warning.


Project Workflow

  1. Collect rumble strip and non-rumble strip images.
  2. Annotate rumble strips using bounding boxes.
  3. Train an RF-DETR Object Detection model.
  4. Evaluate performance on validation data.
  5. Test on unseen highway videos.
  6. Generate real-time warnings.
  7. Create annotated output videos.

Dataset

Dataset Summary

Category Count
Positive Images 46
Negative Images 53
Total Images 99
Classes 1

Class

rumble_strip

Annotation Tool

  • Roboflow

Dataset Characteristics

Positive Images:

  • Highway rumble strips
  • Different lighting conditions
  • Multiple viewpoints

Negative Images:

  • Roads without rumble strips
  • Hard negatives
  • Similar road patterns

Model

Architecture

RF-DETR Object Detection (Small)

Platform

  • Roboflow
  • Python
  • Inference SDK

Input Resolution

512 Γ— 512


Validation Performance

External Validation Results

Metric Score
mAP@50 90.1%
Precision 100.0%
Recall 90.0%
F1 Score 94.7%

Inference Pipeline

Road Video ↓ Frame Extraction ↓ RF-DETR Detection ↓ Confidence Filtering ↓ Temporal Confirmation ↓ Warning Generation ↓ Annotated Video Output


Driver Warning System

The warning system uses:

  • Detection confidence threshold
  • Consecutive detection confirmation
  • Temporal filtering

Warning message:

⚠ RUMBLE STRIP AHEAD

This reduces false positives caused by isolated detections.


External Video Evaluation

Test Video

  • Resolution: 1920 Γ— 804
  • FPS: 30
  • Duration: 5 minutes

Observed Rumble Strip Events

Strip Visibility Window
Strip 1 00:55 – 00:58
Strip 2 02:31 – 02:41
Strip 3 02:52 – 02:53

Results

Detected Events:

3 / 3

Detection Rate:

100%

Warning Lead Time

Strip Warning Time
Strip 1 1.0 sec
Strip 2 3.0 sec
Strip 3 0.6 sec

Example Detection

The system outputs:

  • Bounding boxes
  • Confidence scores
  • Warning overlays
  • Annotated MP4 video

Tech Stack

Machine Learning

  • RF-DETR
  • Computer Vision
  • Object Detection

Tools

  • Roboflow
  • Python
  • OpenCV
  • Inference SDK
  • NumPy

Deployment

  • Roboflow Serverless API

Repository Structure

rumble-strip-detection/
β”‚
β”œβ”€β”€ README.md
β”œβ”€β”€ requirements.txt
β”‚
β”œβ”€β”€ src/
β”‚   β”œβ”€β”€ test_api.py
β”‚   β”œβ”€β”€ video_warning.py
β”‚   └── evaluate_strips.py
β”‚
β”œβ”€β”€ demo/
β”‚   β”œβ”€β”€ rumble_strip_demo.mp4
β”‚   β”œβ”€β”€ screenshot_1.jpg
β”‚   └── screenshot_2.jpg
β”‚
β”œβ”€β”€ docs/
β”‚   └── architecture.png
β”‚
└── results/
    └── v1_results.md

Current Limitations

  • Small dataset size
  • Limited testing videos
  • Detection distance can be improved
  • Occasional false positives
  • Warning lead time varies depending on road geometry

Future Work (V2)

Dataset Improvements

  • Hard-negative mining
  • Additional highway footage
  • More far-distance rumble strip examples

Model Improvements

  • Earlier detection
  • Reduced false positives
  • Increased robustness

System Improvements

  • Distance estimation
  • Time-to-strip estimation
  • Real-time webcam integration
  • Edge deployment

Version History

V1.0

Initial working prototype

Features:

  • 99-image dataset
  • RF-DETR Small
  • 90.1% mAP@50
  • Driver warning prototype
  • Annotated video generation
  • External video testing
  • 3/3 rumble-strip events detected

Author

Hemanath T

Data Science | Machine Learning | Computer Vision

Portfolio: https://hemanath-t.netlify.app/

LinkedIn: www.linkedin.com/in/hemnaththangavel

GitHub: github.com/Hemanath400


About

Computer Vision system for detecting highway rumble strips from forward-facing road videos using RF-DETR, with temporal confirmation and real-time driver warnings.

Topics

Resources

Stars

1 star

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages