Edit model card

Returns facial emotion based on facial human image.

See https://www.kaggle.com/code/dima806/facial-emotions-image-detection-vit for more details.

image/png

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
Downloads last month
19,813
Safetensors
Model size
85.8M params
Tensor type
F32
Β·

Spaces using dima806/facial_emotions_image_detection 11