yashasvini121 / predictive-calc

An interactive web application developed with Streamlit, designed for making predictions using various machine learning models. The app dynamically generates forms and pages from JSON configuration files. ⭐ If you found this helpful, consider starring the repo!
https://predictive-calc.streamlit.app/
MIT License
13 stars 34 forks source link

Facial Emotion Detection #68

Open ananyag309 opened 2 days ago

ananyag309 commented 2 days ago

Problem Description:

Multiclass classification of facial emotions from grayscale images.

MODELS IMPLEMENTED Convolutional Neural Network (for baseline model) MobileNetV2 (for transfer-learning backbone) EfficientNetV2S (for transfer-learning backbone)

LIBRARIES NEEDED Tensorflow Keras Keras_CV Numpy Matplotlib Scikit-learn

Starting with a solid baseline using a CNN, we setup a framework for comparison of different model configurations. Combining data-augmentation with transfer-learning techniques improved performance significantly. Our final model consists of: Preprocessing - Lanczos5 interpolation for resizing and upscaling Data augmentation - Rotation + Horizontal flipping Model (transfer-learning) - EfficientNetV2S backbone for feature-extraction and fine-tuning.

yashasvini121 commented 15 hours ago

Sure, @ananyag309, please proceed!

I assume the final output will be a form where users can upload an image, and the system will return the detected emotion.