Downloads · 30 days
4
33% of all-time downloads
sarasij907/emotion-detection-efficientnet
emotion-detection-efficientnet is a image classification model from sarasij907. Use it when you need a label for an image. It is set up for keras.
ImageNet-pretrained EfficientNetB0 fine-tuned with 5-fold stratified validation. Classes: angry, disgust, fear, happy, neutral, sad, surprise. Mean validation accuracy: 0.5596. Input: RGB image resized to 224x224.
Downloads · 30 days
4
33% of all-time downloads
All-time downloads
12
Public
Repo size
29.1 MB
Likes
0
Public
Click a slice to open those files.
.keras29.1 MB · 100%
From the Hugging Face model README
ImageNet-pretrained EfficientNetB0 fine-tuned with 5-fold stratified validation. Classes: angry, disgust, fear, happy, neutral, sad, surprise. Mean validation accuracy: 0.5596. Input: RGB image resized to 224x224.