dima806's picture
Update README.md
ecab793 verified
|
raw
history blame
769 Bytes
---
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
```