FacialEmotionRecognizer

A simple emotion recognizer project using images classification I did in my journey taking on AI

1.Requirements

git clone https://github.com/maybehieu/FacialEmotionRecognizer.git
pip install -r requirements.txt
python generate_training_data.py -d (~PATH/dataset/) -fer (~PATH/fer2013.csv) -ferplus (~PATH/fer2013new.csv)

• Note: Re-order the label.csv file to /dataset/label/ according to the instruction. (Or you could change the code according to your file structure)

2. Training

python train_mobile.py # for mobilenetv2 implementation

python train_vgg.py # for custom vgg13 implementation

3. Real-time prediction

python frame_pred.py

4. References

• Arxiv - Official paper on FER+ datasets
• Guide and Code - Ideas comes from here
• Mentors: QuangTran & AnNguyen