SafeDrive Model Suite
Models powering the safedrive-ai Python SDK for real-time driver monitoring.
Files
| File | Purpose | Accuracy |
|---|---|---|
yolo_safedrive.pt |
YOLOv8-nano, 9-class detection | mAP50=0.940 |
mobilenet_webcam.pth |
MobileNetV3 eye classifier (webcam fine-tuned) | 97.99% |
mobilenet_best.pth |
MobileNetV3 eye classifier (lab-trained) | 97.99% |
drowsiness_cnn_best.pth |
Custom CNN eye classifier | 96.74% |
face_landmarker.task |
MediaPipe face landmark model | — |
Classes (YOLO)
eye_open, eye_half, eye_closed, mouth_open, mouth_closed, phone, cigarette, seatbelt_on, seatbelt_off
Usage
pip install safedrive-ai
- Downloads last month
- 66