metadata
license: apache-2.0
metrics:
- accuracy
- f1
base_model:
- google/vit-base-patch16-224-in21k
Returns age group with about 91% accuracy based on facial image.
See https://www.kaggle.com/code/dima806/age-group-image-detection-vit for more details.
Classification report:
precision recall f1-score support
MIDDLE 0.8316 0.9278 0.8771 4321
YOUNG 0.9598 0.8563 0.9051 4322
OLD 0.9552 0.9477 0.9515 4322
accuracy 0.9106 12965
macro avg 0.9155 0.9106 0.9112 12965
weighted avg 0.9155 0.9106 0.9112 12965