Edit model card

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
82
Safetensors
Model size
85.8M params
Tensor type
F32
·