Recode-Hive / machine-learning-repos

A curated list of awesome machine learning frameworks, libraries and software (by language). I
MIT License
53 stars 126 forks source link

Enhance Facial Expression Recognition Model with EfficientNet and Advanced Techniques #178

Open bhanushri12 opened 1 month ago

bhanushri12 commented 1 month ago

@sanjay-kv I would like to raise this issue as a part of the GSSoC'24 The current model architecture for facial expression recognition is based on convolutional neural networks (CNNs) designed for image classification tasks. While leveraging pre-trained models from the TIMM library has been effective, there are several ways to further enhance performance and reduce training time. Proposed Improvements

  1. Optimizing the Training Process:

    • Apply regularization techniques like dropout and weight decay to prevent overfitting.
    • Use mixed precision training to enhance training speed and reduce memory usage.
    • Implement differential learning rates for different layers of the network.
  2. Data Augmentation: -Implement a more diverse range of data augmentations to improve model robustness and generalization.

    • Sample augmentations include horizontal flip, rotation, brightness/contrast adjustment, random cropping, Gaussian noise, and Gaussian blur.

I would like to work on this issue as I have already been working with face datasets.

github-actions[bot] commented 1 month ago

Thank you for creating this issue! 🎉 We'll look into it as soon as possible. In the meantime, please make sure to provide all the necessary details and context. If you have any questions reach out to LinkedIn. Your contributions are highly appreciated! 😊

You can also check our CONTRIBUTING.md for guidelines on contributing to this project.

bhanushri12 commented 1 month ago

@sanjay-kv can I start working on this issue?

sanjay-kv commented 4 weeks ago

yes