Edit model card

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
0
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
(1717)
this model

Dataset used to train dima806/fairface_gender_image_detection