Detects gender with about 93.4% accuracy based on an image.

See https://www.kaggle.com/code/dima806/gender-image-classification-vit for details.

image/png

Classification report:

              precision    recall  f1-score   support

      Female     0.9531    0.9051    0.9285      4720
        Male     0.9188    0.9602    0.9391      5280

    accuracy                         0.9342     10000
   macro avg     0.9360    0.9327    0.9338     10000
weighted avg     0.9350    0.9342    0.9341     10000
Downloads last month
706
Safetensors
Model size
85.8M params
Tensor type
F32
·
Inference Examples
Unable to determine this model's library. Check the docs .

Model tree for dima806/fairface_gender_image_detection

Finetuned
(1784)
this model

Dataset used to train dima806/fairface_gender_image_detection