illimax/jolgsm-distraction-detector

JolGSM Distraction Detector (RT-DETR-L)

운전자 부주의/졸음 감지를 위한 RT-DETR-L 파인튜닝 모델입니다.

클래스

ID	클래스	설명
0	face_normal	정상 주시 얼굴
1	face_distracted	부주의 상태 얼굴 (통화 재현)
2	phone	휴대폰

학습 정보

• 베이스 모델: RT-DETR-L (rtdetr-l.pt)
• 데이터셋: AI-Hub 졸음운전 예방을 위한 운전자 상태 정보 영상 (통제환경 bbox)
• 학습 이미지: 33,519장 (train) / 4,335장 (val)
• 입력 해상도: 640×640
• 학습 epochs: 20

사용법

from ultralytics import RTDETR

model = RTDETR("jolgsm-distraction-detector.pt")
results = model.predict("image.jpg", conf=0.65)
results[0].show()

프로젝트

JolGSM GitHub