我正在使用来自的FER2013Plus数据集 https://github.com/Microsoft/FERPlus 其中包含fer2013new.csv文件。此文件包含数据集中每个图像的标签。标签上的示例可以是:
每个维度都是不同的情感。最后,在他们的论文中 https://arxiv.org/pdf/1608.01041.pdf ,他们将标签分布转换为概率=>新标签将成为
(0.5, 0, 0, 0.25, 0.125, 0, 0, 0.375)
现在我在训练的时候 tf.nn.softmax_cross_entropy_with_logits_v2
tf.nn.softmax_cross_entropy_with_logits_v2
非常感谢您的帮助!!
以下是论文节选:
“我们把大多数的情感看作是单一的 情绪标签,我们衡量预测的准确性 大多数人的情绪。”
他们使用的是离散分类任务。所以你只需要 tf.argmax() tf.argmax()
tf.argmax()
例如,如果您的标签是 (0.5, 0, 0, 0.25, 0.125, 0, 0, 0.375)