Returns age group 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.8162    0.9250    0.8672      1080
       YOUNG     0.9583    0.8511    0.9015      1081
         OLD     0.9537    0.9334    0.9434      1081

    accuracy                         0.9031      3242
   macro avg     0.9094    0.9032    0.9040      3242
weighted avg     0.9094    0.9031    0.9041      3242

Downloads last month: 6,258

Safetensors

Model size

85.8M params

Tensor type

F32

Inference API

Image Classification

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.

dima806
/

faces_age_detection

Spaces using dima806/faces_age_detection 3