Downloads · 30 days
4.5K
0% of all-time downloads
facial_emotions_image_detection
dima806/facial_emotions_image_detection
facial_emotions_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.
Returns facial emotion with about 91% accuracy based on facial human image.
All-time downloads
2.5M
Public
Parameters
85.8M
7.6 GB on disk
Likes
129
Public
Repo makeup
Click a slice to open those files.
.pt2.7 GB · 53%
At a glance
- Task
- Image Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- vit
- Access
- Public
- Created
- Oct 2, 2023
- Updated
- Oct 19, 2024
- SHA
- 747cf166
Base models
Model card
From the Hugging Face model README
Returns facial emotion with about 91% accuracy based on facial human image.
See https://www.kaggle.com/code/dima806/facial-emotions-image-detection-vit for more details.
Classification report:
precision recall f1-score support
sad 0.8394 0.8632 0.8511 3596
disgust 0.9909 1.0000 0.9954 3596
angry 0.9022 0.9035 0.9028 3595
neutral 0.8752 0.8626 0.8689 3595
fear 0.8788 0.8532 0.8658 3596
surprise 0.9476 0.9449 0.9463 3596
happy 0.9302 0.9372 0.9336 3596
accuracy 0.9092 25170
macro avg 0.9092 0.9092 0.9091 25170
weighted avg 0.9092 0.9092 0.9091 25170
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- License
- apache-2.0
- Hf Inference
- live
- Created
- Oct 2, 2023
- Updated
- Oct 19, 2024