dima806's picture
Update README.md
cb6ed8d verified
|
raw
history blame
1.17 kB
---
license: apache-2.0
metrics:
- accuracy
- f1
base_model:
- google/vit-base-patch16-224-in21k
pipeline_tag: image-classification
library_name: transformers
---
Detects age group with about 18% accuracy based on an image.
See https://www.kaggle.com/code/dima806/age-group-image-classification-vit for details.
![image/png](/static-proxy?url=https%3A%2F%2Fcdn-uploads.huggingface.co%2Fproduction%2Fuploads%2F6449300e3adf50d864095b90%2FsG1GZFgo5TlzzFqal5obl.png%3C%2Fspan%3E)%3C!-- HTML_TAG_END -->
```
Classification report:
precision recall f1-score support
0-2 0.0000 0.0000 0.0000 0
3-9 0.1667 0.2500 0.2000 8
10-19 0.1667 0.0769 0.1053 13
20-29 0.4000 0.3243 0.3582 37
30-39 0.0000 0.0000 0.0000 18
40-49 0.0000 0.0000 0.0000 13
50-59 0.1579 0.5000 0.2400 6
60-69 0.0000 0.0000 0.0000 3
more than 70 0.0000 0.0000 0.0000 2
accuracy 0.1800 100
macro avg 0.0990 0.1279 0.1004 100
weighted avg 0.1925 0.1800 0.1766 100
```