|
--- |
|
license: apache-2.0 |
|
metrics: |
|
- accuracy |
|
- f1 |
|
base_model: |
|
- google/vit-base-patch16-224-in21k |
|
--- |
|
Returns with about 98.7% accuracy whether the face belongs to man or woman based on face image. |
|
|
|
See https://www.kaggle.com/code/dima806/man-woman-face-image-detection-vit for more details. |
|
|
|
![image/png](https://cdn-uploads.huggingface.co/production/uploads/6449300e3adf50d864095b90/t9MgehgAAZEJAXOebhfjO.png) |
|
|
|
``` |
|
Classification report: |
|
|
|
precision recall f1-score support |
|
|
|
man 0.9885 0.9857 0.9871 51062 |
|
woman 0.9857 0.9885 0.9871 51062 |
|
|
|
accuracy 0.9871 102124 |
|
macro avg 0.9871 0.9871 0.9871 102124 |
|
weighted avg 0.9871 0.9871 0.9871 102124 |
|
``` |