- [train_emotions-pytorch.ipynb](affectnet/train_emotions-pytorch.ipynb): the **main file** in this repository. It contains the training of our EmotiEffNet models, such as [EmotiEffNet-B2](../models/affectnet_emotions/enet_b2_8_best.pt) and classification of emotional features for AFEW and VGAF datasets. It is necessary to run preprocessing of the datasets from train_emotions.ipynb, AFEW_train.ipynb and VGAF_train.ipynb
0 commit comments