Skip to content

yoshidan/pytorch-eyeblink-detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pytorch-eyeblink-detection

Simple Eye Blink Detection with CNN

  • Using Mini Xception
  • Model size is about 230KB

Trained Model

Trained by CEW dataset.

  • Validation Accuracy : 96.7%
  • Validation Loss : 0.1256

Here is the result of sample image.
| Image | Open Probability | Guided Backprop | Grad-Cam | Guided Grad-Cam

Retrain

cd src
python train.py
python check.py  #check.py supports cpu only

Reference

Releases

No releases published

Packages

No packages published

Languages