운전자 피로도 분석 시스템 - 영상 기반 눈 감김 분석을 통한 실시간 피로도 판단 라이브러리
Project description
🚗 SleepyDriver - AI 기반 졸음 감지 라이브러리
SleepyDriver는 실시간 비디오에서 운전자의 졸음 상태를 AI로 감지하는 파이썬 라이브러리입니다. 5가지 다양한 감지 모델을 지원하며, 간단한 API로 프로젝트에 쉽게 통합할 수 있습니다.
✨ 주요 기능
- 🎯 5가지 감지 모델: OpenCV, 머신러닝(RF), 딥러닝(CNN), MediaPipe, 적외선 기반
- ⚡ 실시간 처리: 웹캠에서 실시간 졸음 감지 (~30 FPS)
- 🔧 간단한 API: 3줄의 코드로 졸음 감지 시스템 구축
- 📦 플러그인 아키텍처: 사용자 정의 모델 쉽게 추가 가능
- 🛡️ 안정성: 강력한 에러 처리와 의존성 관리
- 🖥️ CLI 지원: 설치 후 바로 사용 가능한 명령행 도구
🚀 빠른 시작
설치
# 기본 설치 (OpenCV 모델만)
pip install sleepy-driver
# 모든 모델 포함 설치 (권장)
pip install sleepy-driver[all]
# 선택적 설치
pip install sleepy-driver[ml] # 머신러닝 모델
pip install sleepy-driver[dl] # 딥러닝 모델
CLI로 바로 사용
# 기본 실행
sleepy-driver-demo
# 다른 모델로 실행
sleepy-driver-demo --model mlp --threshold 2000
# 사용 가능한 모델 확인
sleepy-driver-demo --list-models
코드로 사용 (초간단!)
🚀 원라이너 (가장 간단!)
# 1줄로 바로 시작!
from sleepy_driver import start_detection
start_detection('mlp') # MLP 모델로 바로 시작
start_detection('ml', threshold_ms=2000) # ML 모델, 2초 임계값
start_detection('opencv') # OpenCV 모델
start_detection('point') # MediaPipe 모델
start_detection('infrared') # 적외선 모델
📋 지원 모델
| 모델 | 설명 | 장점 | 의존성 |
|---|---|---|---|
| opencv | OpenCV 기반 전통적 컴퓨터 비전 | 빠름, 의존성 적음 | 없음 |
| ml | RandomForest 머신러닝 | 균형잡힌 성능 | scikit-learn |
| mlp | CNN 딥러닝 | 높은 정확도 | PyTorch |
| point | MediaPipe 랜드마크 | 실시간성 우수 | 없음 |
| infrared | 적외선 CNN 딥러닝 | 야간 감지 우수 | PyTorch |
📊 성능 벤치마크
| 모델 | 평균 FPS | 정확도 | 메모리 사용량 |
|---|---|---|---|
| OpenCV | ~35 FPS | 85% | ~50MB |
| ML (RF) | ~30 FPS | 90% | ~100MB |
| MLP (CNN) | ~28 FPS | 95% | ~200MB |
| Point | ~40 FPS | 80% | ~30MB |
| Infrared | ~25 FPS | 92% | ~250MB |
테스트 환경: MacBook Pro M1, 720p 웹캠
👥 팀원 소개
운전자 피로도 분석 시스템 - 영상 기반 눈 감김 분석을 통한 실시간 피로도 판단 라이브러리
🎯 프로젝트 목적
- 운전자의 눈이 감겼는가를 이진 분류
- 눈을 감은 시간이 몇 초인지 정량 분석
- 실시간 운전자 피로도 분석 모듈의 통합 라이브러리 구축
- 스마트카 연계: ADAS 및 자율주행 시스템의 운전자 모니터링 기능 강화
- 사회적 효과: 사고 예방 AI 구현 가능
- 기술 학습 효과: 실시간 영상 처리, 시계열 데이터 분석, Vision 기반 분류 등 산업 적용력 높은 실전형 경험
👨💻 개발 팀
| 이름 | 역할 | GitHub |
|---|---|---|
| 김진현 | AIHub 기반 딥러닝 모델 설계 및 적외선 딥러닝 라벨 정리, PyTorch GPU 리서치, 적외선 모델 구현 및 학습, 오류 디버깅 지원 | @JinhyeonK |
| 김호탁 | 전처리 모듈 구축 및 구조 기획, 좌표 기반 눈 감김 판별 알고리즘 구현, 딥러닝 모델 학습 및 통합, 상속 구조 정리 및 패키지화 주도 | @kimhotac |
| 박준규 | OpenCV 기반 접근법 개발, 디렉토리 정리 및 구조 패키지화, 졸음 검출 모듈 구현, PyPI 라이브러리 배포 주도 | @ParkJunGyu26 |
| 안승현 | Mediapipe 기반 좌표 추출 및 눈 감김 판별 알고리즘 구현, Kaggle 데이터 기반 딥러닝 모델 개발 및 최적화, PyTorch GPU 환경 세팅 및 적외선 모델 학습 참여 | @asho227 |
| 윤선아 | 데이터 수집 및 머신러닝 모델 (SVM, XGBoost, RandomForest) 개발 전반 수행, 모델 최적화 및 적용 | @dotoriysa |
🛠️ 개발자 가이드
로컬 개발 설정
# 저장소 클론
git clone https://github.com/kimhotac/sleepy-driver-detector.git
cd sleepy-driver-detector
# 개발 의존성 설치
pip install -e .[dev]
# 테스트 실행
pytest tests/
# 코드 포맷팅
black sleepy_driver/
flake8 sleepy_driver/
패키지 빌드
# 빌드 도구 설치
pip install build twine
# 패키지 빌드
python -m build
# PyPI 업로드 (관리자만)
twine upload dist/*
🤝 기여하기
- Fork 저장소
- 기능 브랜치 생성 (
git checkout -b feature/amazing-feature) - 변경사항 커밋 (
git commit -m 'Add amazing feature') - 브랜치 푸시 (
git push origin feature/amazing-feature) - Pull Request 생성
📄 라이선스
이 프로젝트는 MIT 라이선스 하에 배포됩니다. 자세한 내용은 LICENSE 파일을 참조하세요.
🙏 감사의 말
- MediaPipe - 얼굴 랜드마크 감지
- OpenCV - 컴퓨터 비전 라이브러리
- PyTorch - 딥러닝 프레임워크
- scikit-learn - 머신러닝 라이브러리
🆘 지원 및 문의
- 📖 문서: GitHub README
- 🐛 버그 리포트: GitHub Issues
- 💬 디스커션: GitHub Discussions
- 📧 이메일: junju404@naver.com (박준규)
⚠️ 주의사항: 이 라이브러리는 보조 도구로만 사용하세요. 실제 운전 시에는 항상 안전을 최우선으로 하고, 졸음을 느끼면 즉시 안전한 곳에 정차하여 휴식을 취하세요.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file sleepy_driver-1.2.4.tar.gz.
File metadata
- Download URL: sleepy_driver-1.2.4.tar.gz
- Upload date:
- Size: 9.7 MB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.1.0 CPython/3.9.23
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
fffeaa85bea66c8ce6593288d4027080ffe0979f5d835b684dc09a4f26be6a6c
|
|
| MD5 |
2c7ffc401abe755eaddeedd57389b732
|
|
| BLAKE2b-256 |
3e01da21df165291700b6cdff6176cadf06db293b1c8c19320d5fbb3fce92be6
|
File details
Details for the file sleepy_driver-1.2.4-py3-none-any.whl.
File metadata
- Download URL: sleepy_driver-1.2.4-py3-none-any.whl
- Upload date:
- Size: 9.7 MB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.1.0 CPython/3.9.23
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
66b46060ec75ec4a8d2772e70fa967c331b0db92868a09add9c0a50d30386825
|
|
| MD5 |
1dc04c54b4ec210146cc4bd8ffeedbbd
|
|
| BLAKE2b-256 |
968a35502c1ac4054bdb0879edb11190d0e933c3bf080fe62e0f8f0e0c8fe39a
|