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.

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

Downloads last month: 44,610

Safetensors

Model size

85.8M params

Tensor type

F32

Model tree for dima806/man_woman_face_image_detection

Base model

google/vit-base-patch16-224-in21k

Finetuned

(2544)

this model

dima806
/

man_woman_face_image_detection

Model tree for dima806/man_woman_face_image_detection

Spaces using dima806/man_woman_face_image_detection 3