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: 49,456

Safetensors

Model size

85.8M params

Tensor type

F32

Inference Examples

Image Classification

This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Model tree for dima806/man_woman_face_image_detection

Base model

google/vit-base-patch16-224-in21k

Finetuned

(1701)

this model