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.

image/png

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
Downloads last month
18,757
Safetensors
Model size
85.8M params
Tensor type
F32
Β·
Inference Examples
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/faces_age_detection

Finetuned
(1770)
this model

Spaces using dima806/faces_age_detection 4