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

Downloads last month: 4,694

Safetensors

Model size

85.8M params

Tensor type

F32

Inference Providers NEW

Image Classification

This model is not currently available via any of the supported third-party Inference Providers, and the model is not deployed on the HF Inference API.

Model tree for dima806/faces_age_detection

Base model

google/vit-base-patch16-224-in21k

Finetuned

(1840)

this model

dima806
/

faces_age_detection

Model tree for dima806/faces_age_detection

Spaces using dima806/faces_age_detection 4