|
--- |
|
license: apache-2.0 |
|
metrics: |
|
- accuracy |
|
- f1 |
|
base_model: |
|
- google/vit-base-patch16-224-in21k |
|
pipeline_tag: image-classification |
|
library_name: transformers |
|
datasets: |
|
- nateraw/fairface |
|
--- |
|
Detects age group with about 59% accuracy based on an image. |
|
|
|
See https://www.kaggle.com/code/dima806/age-group-image-classification-vit for details. |
|
|
|
 |
|
|
|
``` |
|
Classification report: |
|
|
|
precision recall f1-score support |
|
|
|
0-2 0.7803 0.7500 0.7649 180 |
|
3-9 0.7998 0.7998 0.7998 1249 |
|
10-19 0.5361 0.4236 0.4733 1086 |
|
20-29 0.6402 0.7221 0.6787 3026 |
|
30-39 0.4935 0.5083 0.5008 2099 |
|
40-49 0.4848 0.4386 0.4606 1238 |
|
50-59 0.5000 0.4814 0.4905 725 |
|
60-69 0.4497 0.4685 0.4589 286 |
|
more than 70 0.6897 0.1802 0.2857 111 |
|
|
|
accuracy 0.5892 10000 |
|
macro avg 0.5971 0.5303 0.5459 10000 |
|
weighted avg 0.5863 0.5892 0.5844 10000 |
|
``` |
