A newer version of the Gradio SDK is available:
6.5.1
metadata
title: Driver Drowsiness Detection
emoji: 🚗
colorFrom: blue
colorTo: indigo
sdk: gradio
sdk_version: 5.31.0
app_file: app.py
pinned: false
license: mit
paper: arxiv.org/abs/2505.17392
Driver Drowsiness Detection System
This is a real-time driver drowsiness detection system that uses computer vision and deep learning to detect signs of drowsiness in drivers. The system can process webcam feeds, video files, and single images.
Features
- Real-time webcam monitoring
- Video file processing
- Single image analysis
- Face detection and drowsiness prediction
- Visual feedback with bounding boxes and status indicators
How to Use
- Webcam Mode: Click the "Start Webcam" button to begin real-time monitoring
- Video Mode: Upload a video file for processing
- Image Mode: Upload a single image for analysis
The system will display the results with:
- Green box: Alert (not drowsy)
- Red box: Drowsy
- Probability score for drowsiness
Technical Details
- Built with PyTorch and Vision Transformer (ViT)
- Uses OpenCV for face detection
- Gradio interface for easy interaction
- Real-time processing capabilities
Model
The system uses a Vision Transformer (ViT) model trained on driver drowsiness detection. The model is capable of detecting subtle signs of drowsiness in facial expressions.