Downloads · 30 days
247
5% of all-time downloads
face_emotions_image_detection
dima806/face_emotions_image_detection
face_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.
Predicts face emotion based on facial image.
All-time downloads
4.9K
Public
Parameters
85.8M
10.3 GB on disk
Likes
6
Public
Repo makeup
Click a slice to open those files.
.pt6.2 GB · 64%
At a glance
- Task
- Image Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- vit
- Access
- Public
- Created
- Dec 16, 2023
- Updated
- Oct 19, 2024
- SHA
- a89ee07e
Base models
Spaces
Model card
From the Hugging Face model README
Predicts face emotion based on facial image.
See https://www.kaggle.com/code/dima806/face-emotions-image-detection-vit for more details.
Classification report:
precision recall f1-score support
Ahegao 0.9738 0.9919 0.9828 1611
Angry 0.8439 0.6580 0.7394 1611
Happy 0.8939 0.9261 0.9098 1611
Neutral 0.6056 0.7635 0.6755 1611
Sad 0.6661 0.5140 0.5802 1611
Surprise 0.7704 0.8733 0.8186 1610
accuracy 0.7878 9665
macro avg 0.7923 0.7878 0.7844 9665
weighted avg 0.7923 0.7878 0.7844 9665
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- License
- apache-2.0
- Created
- Dec 16, 2023
- Updated
- Oct 19, 2024