WuJie1010 / Facial-Expression-Recognition.Pytorch

A CNN based pytorch implementation on facial expression recognition (FER2013 and CK+), achieving 73.112% (state-of-the-art) in FER2013 and 94.64% in CK+ dataset
MIT License
1.78k stars 545 forks source link

Hi, can i train a binary problem? #83

Closed Angel113110 closed 4 years ago

Angel113110 commented 4 years ago

Dear author, thanks for sharing so wonderful work. I have a question to ask you. In my case, I only focus on smile expression. Can I train a binary problem based on your work? Can you give some help or advice? Thanks!

WuJie1010 commented 4 years ago

In my opinion, you can separate the smile data and other data into two categories. In addition, you should add more neutral facial expressions. You may encounter data imbalance. Remember to use some techniques of data imbalance.