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

Returns with about 98.7% accuracy whether the face belongs to man or woman based on face image.

See https://www.kaggle.com/code/dima806/man-woman-face-image-detection-vit for more details.

image/png

Classification report:

              precision    recall  f1-score   support

         man     0.9885    0.9857    0.9871     51062
       woman     0.9857    0.9885    0.9871     51062

    accuracy                         0.9871    102124
   macro avg     0.9871    0.9871    0.9871    102124
weighted avg     0.9871    0.9871    0.9871    102124