yeeeengyu/YOLO-drowsiness-detector

YOLO Drowsiness Detector for LightBox

LightBox에서 사용하는 운전자 졸음감지 모델입니다.
졸음 자체는 mediapipe로 계산하며, 이 모델은 운전자 얼굴 / 눈을 감지하는 모델입니다.

Model Details

• Architecture: YOLO

• Framework: Ultralytics / PyTorch

• Task: Object Classification

• Purpose: Driver drowsiness detection

• environments: RTX 4090 + Cuda 12.4 + Ubuntu 22.04 LTS

Classes

• open
• closed
• face