An open API service indexing awesome lists of open source software.

https://github.com/mosesab/yolov11-drowsiness-detection

This repo contains the fine-tune implementation of a drowsy detection system, using a YOLO v11 model.
https://github.com/mosesab/yolov11-drowsiness-detection

computer-vision drowsiness-detection drowsiness-detection-python fine-tuning yolo yolov11

Last synced: about 1 month ago
JSON representation

This repo contains the fine-tune implementation of a drowsy detection system, using a YOLO v11 model.

Awesome Lists containing this project

README

          

### YOLO v11 for Driver Drowsiness Detection

📌 **Notebook moved to Kaggle:**
👉 https://www.kaggle.com/code/themosesab/yolov11-for-driver-drowsiness-detection

This repository contains a complete, end-to-end pipeline for training a YOLO v11 classification model to detect driver drowsiness using the ultralytics library.
The project covers every stage of the machine learning lifecycle, from automated data acquisition and cleaning to model evaluation and explainability using GradCAM.

![alt text](https://img.shields.io/badge/Python-3.10-blue.svg)

![alt text](https://img.shields.io/badge/Framework-PyTorch-orange.svg)

![alt text](https://img.shields.io/badge/License-MIT-green.svg)

![alt text](https://img.shields.io/badge/Data-Kaggle-blue.svg)

### Quick Summary
Accuracy 99.80% Overall correctness on the test set.
APCER 0.00% Rate of 'Drowsy' drivers missed (False Negatives).
BPCER 0.41% Rate of 'Non Drowsy' drivers flagged (False Positives).
ACER 0.21% Average of APCER and BPCER.

### Acknowledgements

This project uses the following datasets from Kaggle:
- Driver Drowsiness Dataset (DDD) by ISMAIL NASRI.
- Drowsy Detection Dataset by YASHAR JEBRAEILY.

### License
This project is licensed under the MIT License. See the LICENSE file for details.