Python module which detects if eye is open or closed.
This project uses Closed Eyes In The Wild (CEW) dataset.
The training program expects two pickle files inside dataset_dir(option for training program).
This two files are train.pkl and test.pkl which contains file locactions to training and test sets images respectively.
The following program can be used to split dataset into train and test sets and save the file locations in respective pickle files.
python -m preprocess [options]
- --dataset_dir - Directroy which contains
ClosedFacefolder andOpenFacefolder.
Training program expects dataset dir which contains ClosedFace and OpenFace folders, and train.pkl and test.pkl files.
The following program can be used to train the model.
python -m train [options]