Skip to content

Tandon-A/Drowsiness-Detection-Mediapipe

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Drowsiness-Detection-Mediapipe

The code uses MediaPipe model for detecting a face in the video stream.

Features based on facial landmark positions are calculated and are used by a LSTM model to predict whether the use is alert or drowsy.

As a first step, calibration is performed, which calculates the facial features over a small number of frames. The user is reminded to be in neutral position in this process.

Calibration returns the mean and standard deviation values to normalize the facial features at inference step.

Fig 1: Calibration

During inference, first the features are calculated and are normalized. These features are sent over to the LSTM model to make a prediction. The LSTM model computes its prediction over five continuous frames and returns the predicted state.

Fig 2: Alert State

Fig 3: Drowsy State

Fig 4: Demo inference

Projects

Sleepless Academy

We apply the idea of drowsiness detection to a virtual classroom / online MOOC platform setting. The WebApp shows a demo where the student is alerted when drowsy. The statistics obtained over the course can then help course content creators to improve their offerings.

LSTM Model

Fig 5: Web Application Demo

Authors

Releases

No releases published

Packages

No packages published

Languages