search

ArnabKumarRoy02 / ResEmoteNet

This repository is the official code for ResEmoteNet. The project is written in Python using PyTorch in MacBook Pro (M2 Pro 10-core CPU and 16-core GPU).
MIT License
17 stars 4 forks source link

Clarification on Expression Columns in Your Code #11

Closed moonyy929 closed 1 week ago

moonyy929 commented 1 week ago

Hi, Arnab. Thanks for your great work! I would like to ask about the dataset that your code is based on, particularly regarding the expression columns. For example, in the code, the header is defined as class_labels = ['happiness', 'surprise', 'sadness', 'anger', 'disgust', 'fear', 'neutral']. I would like to know which dataset corresponds to these columns.

Thank you!

ArnabKumarRoy02 commented 1 week ago

Hi @moonyy929, thanks for pointing out the wrong labels. It's been updated. The labels are for all the datasets that have been used in the project: FER2013, RAF-DB, AffectNet-7.