这个基于wav2vec2的情绪检测模型是在 emo-DB dataset 上训练的。训练代码可以在 here 中找到。

情绪类别如下：

• 0: 'anger'
• 1: 'boredom'
• 2: 'disgust'
• 3: 'fear'
• 4: 'happiness'
• 5: 'sadness'
• 6: 'neutral'

inference.py展示了如何使用该模型。