search

phamquiluan / ResidualMaskingNetwork

ICPR 2020: Facial Expression Recognition using Residual Masking Network
https://ieeexplore.ieee.org/document/9411919
MIT License
457 stars 93 forks source link

How to use it to train more or less than 7 emotions? #12

Closed srushti108 closed 3 years ago

srushti108 commented 4 years ago

Hii, I tried your's model for prediction , it gives much better result for the 7 classes in fer2013. But, I want to use it to train on more 5 emotion classes . For now, I was trying with one less class, I did conversion of my 6 classes data to the fer data format and make change num_classes=6 in fer2013_config and tried with default 'alexnet' model. Training is on, till now it's running without error. But, is that sufficient changes? Or Is it possible to train for more classes in this way? (Is there any changes require in tta_trainer.py or alexnet.py ?) Please help me to clarify with this.

phamquiluan commented 3 years ago

I did exp in this case yet, but, it should be ok. if it's not, try to update the last fully-connected layer configuration :pray: