Detects gender with about 93.4% accuracy based on an image.
See https://www.kaggle.com/code/dima806/gender-image-classification-vit for details.
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
Model tree for dima806/fairface_gender_image_detection
Base model
google/vit-base-patch16-224-in21k