Edit model card

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
Downloads last month
218,116
Safetensors
Model size
85.8M params
Tensor type
F32
·
Inference API
Drag image file here or click to browse from your device

Model tree for dima806/man_woman_face_image_detection

Finetuned
(1668)
this model