dima806's picture
Update README.md
8a2180a verified
metadata
license: apache-2.0
metrics:
  - accuracy
  - f1
base_model:
  - google/vit-base-patch16-224-in21k

Returns whether there is an open or a closed eye given an image from surrounding area with about 99% accuracy.

See https://www.kaggle.com/code/dima806/closed-eye-image-detection-vit for more details.

Classification report:

              precision    recall  f1-score   support

    closeEye     0.9921    0.9888    0.9904      4296
     openEye     0.9889    0.9921    0.9905      4295

    accuracy                         0.9905      8591
   macro avg     0.9905    0.9905    0.9905      8591
weighted avg     0.9905    0.9905    0.9905      8591