Edit model card

Returns age bin based on a facial image.

See https://www.kaggle.com/code/dima806/facial-age-years-detection-vit for more details.

image/png

Classification report:

              precision    recall  f1-score   support

          01     0.7341    0.9056    0.8109       445
          02     0.4494    0.6787    0.5407       445
          03     0.6978    0.2854    0.4051       445
          04     0.8421    0.1438    0.2457       445
          05     0.5707    0.9618    0.7163       445
       06-07     0.7030    0.5798    0.6355       445
       08-09     0.6500    0.8180    0.7244       445
       10-12     0.6993    0.7056    0.7025       445
       13-15     0.8034    0.7438    0.7725       445
       16-20     0.7006    0.7416    0.7205       445
       21-25     0.6796    0.6292    0.6534       445
       26-30     0.4241    0.5843    0.4915       445
       31-35     0.4654    0.2270    0.3051       445
       36-40     0.4606    0.3416    0.3923       445
       41-45     0.5074    0.6944    0.5863       445
       46-50     0.4896    0.5811    0.5314       444
       51-55     0.5158    0.5506    0.5326       445
       56-60     0.5000    0.3491    0.4111       444
       61-65     0.7083    0.1910    0.3009       445
       66-70     0.4778    0.7995    0.5981       444
       71-80     0.7687    0.7169    0.7419       445
       81-90     0.8425    0.9978    0.9136       445
         90+     0.9978    1.0000    0.9989       444

    accuracy                         0.6185     10231
   macro avg     0.6386    0.6185    0.5970     10231
weighted avg     0.6386    0.6185    0.5970     10231
Downloads last month
3,799
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/facial_age_image_detection

Finetuned
(1691)
this model

Spaces using dima806/facial_age_image_detection 6