human-emotion-recognition

Human Emotion Recognition

Deep learning models for classifying human facial emotions.

Emotion Classes

• 😠 Angry
• 😨 Fear
• 😊 Happy
• 😢 Sad
• 😲 Surprise

Model Performance

🏆 Best model: Base CNN with 92.41% test accuracy

Models

Usage

Python

from huggingface_hub import hf_hub_download
import tensorflow as tf
import numpy as np

# Download model
model_path = hf_hub_download(
    repo_id="dafisnadhif/human-emotion-recognition",
    filename="model_base.h5"
)

# Load model
model = tf.keras.models.load_model(model_path)

# Predict
CLASS_NAMES = ['Angry', 'Fear', 'Happy', 'Sad', 'Surprise']
predictions = model.predict(img_batch)
print(CLASS_NAMES[np.argmax(predictions[0])])

TensorFlow Lite

from huggingface_hub import hf_hub_download
import tensorflow as tf

tflite_path = hf_hub_download(
    repo_id="dafisnadhif/human-emotion-recognition",
    filename="tflite/best_model.tflite"
)

interpreter = tf.lite.Interpreter(model_path=tflite_path)
interpreter.allocate_tensors()

TensorFlow.js

const model = await tf.loadLayersModel(
  'https://huggingface.co/dafisnadhif/human-emotion-recognition/resolve/main/tfjs_model/model.json'
);

Training Details

License

MIT License

Author

Dafis Nadhif Saputra