Downloads · 30 days
50.1K
53% of all-time downloads
skin_types_image_detection
dima806/skin_types_image_detection
skin_types_image_detection is a image classification model from dima806. Use it when you need a label for an image. It is set up for transformers. The card lists the license as apache-2.0.
Detects the skin type (dry, normal, oily) based on facial image.
All-time downloads
94.4K
Public
Parameters
85.8M
4.1 GB on disk
Likes
17
Public
Repo makeup
Click a slice to open those files.
.pt2.7 GB · 62%
At a glance
- Task
- Image Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- vit
- Access
- Public
- Created
- Feb 24, 2024
- Updated
- Oct 19, 2024
- SHA
- 5aace4f4
Base models
Model card
From the Hugging Face model README
Detects the skin type (dry, normal, oily) based on facial image.
See https://www.kaggle.com/code/dima806/skin-types-image-detection-vit for details.
Classification report:
precision recall f1-score support
dry 0.6829 0.6346 0.6578 509
normal 0.6414 0.6314 0.6364 510
oily 0.6390 0.6941 0.6654 510
accuracy 0.6534 1529
macro avg 0.6544 0.6534 0.6532 1529
weighted avg 0.6544 0.6534 0.6532 1529
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- License
- apache-2.0
- Created
- Feb 24, 2024
- Updated
- Oct 19, 2024