Downloads · 30 days
177
5% of all-time downloads
hair_type_image_detection
dima806/hair_type_image_detection
hair_type_image_detection is a image classification model from dima806. Use it when you need a label for an image. The card lists the license as apache-2.0.
Returns the hair type (with 93% accuracy) based on facial image.
All-time downloads
3.7K
Public
Parameters
85.8M
1 GB on disk
Likes
5
Public
Repo makeup
Click a slice to open those files.
.pt687 MB · 50%
At a glance
- Task
- Image Classification
- License
- apache-2.0
- Model type
- vit
- Access
- Public
- Created
- Oct 15, 2024
- Updated
- Oct 19, 2024
- SHA
- c918d585
Base models
Spaces
Model card
From the Hugging Face model README
Returns the hair type (with 93% accuracy) based on facial image.
See https://www.kaggle.com/code/dima806/hair-type-image-detection-vit for details.
Classification report:
precision recall f1-score support
curly 0.9109 0.8932 0.9020 206
dreadlocks 0.9667 0.9902 0.9783 205
kinky 0.9423 0.9561 0.9492 205
straight 0.9585 0.8981 0.9273 206
wavy 0.8651 0.9029 0.8836 206
accuracy 0.9280 1028
macro avg 0.9287 0.9281 0.9281 1028
weighted avg 0.9287 0.9280 0.9280 1028
- Task
- Image Classification
- Type
- vit
- License
- apache-2.0
- Created
- Oct 15, 2024
- Updated
- Oct 19, 2024