Skip to content

jasperfue/AIIM

Repository files navigation

Face Image Based Emotion Detection with Neural Networks

This repository contains the code and resources used for our project on face image-based emotion detection using neural networks. This project was conducted as part of the "Artificial Intelligence and Information Management" lecture, culminating in a Kaggle competition where we secured second place with a weighted F1-score of approximately 0.855.

Repository Structure

  • notebook.ipynb: This notebook contains the method that achieved our best result in the competition. It includes the final model and the data preprocessing and augmentation techniques used.
  • scratch-model.ipynb: This notebook shows our custom-designed Convolutional Neural Network (CNN) built from scratch. It details the architecture and training process of our initial model.
  • data-analysis.ipynb: This notebook focuses on the analysis and augmentation of the data. It includes exploratory data analysis (EDA) and the various data augmentation techniques applied to improve model performance.
  • oversampling.ipynb: This notebook demonstrates our approach to handling class imbalance through oversampling and the use of stratified splits. It details the methods and their impact on model performance.
  • ensemble.ipynb: This notebook explores future improvements of the model using ensemble learning techniques. It includes preliminary implementations and discusses potential benefits.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published