This project applies machine learning to detect facial emotions in Alzheimer’s patients, with the goal of aiding in their care and understanding their emotional states.
Ensure you have Python installed, then set up your environment with the required libraries:
• PyTorch: For creating deep learning models.
• EfficientNet PyTorch: An efficient model for deep learning.
• TorchSampler: For imbalanced dataset handling.
pip install torch torchvision efficientnet_pytorch torchsampler– Installs PyTorch, torchvision, EfficientNet, and TorchSampler.
pip install fer– Installs the FER library for facial emotion recognition.pip install ffmpeg moviepy– Installs ffmpeg and moviepy, necessary for video processing.
The code includes data augmentation techniques to improve model robustness, avoiding bias by excluding ‘ColorJitter’.
The project uses the FER library to detect emotions in images, presenting the detected emotions with bounding boxes.
This project welcomes feedback, corrections, and collaborations to improve and expand its impact.
For detailed instructions, please refer to the comments within the code in the Jupyter Notebook.