字符识别模型可能过拟合?

时间:2020-02-17 16:25:47

标签: python tensorflow machine-learning keras computer-vision

我正在尝试训练模型以识别使用魔杖绘制的手势,该魔杖由IR摄像机拾取并馈入模型中。 我使用A-Z手写字符数据集以及我训练过的两个其他自定义符号训练了模型。 当前,当直接绘制所有符号时,它会正确检测所有符号,但是如果将任何随机符号/图形放入模型中,它也会将它们标识为我添加的两个自定义符号之一。 这是我的困惑矩阵: enter image description here

有什么想法吗? 谢谢!

1 个答案:

答案 0 :(得分:0)

好吧,由于您的模型必须将任何给定的输入分类到任何一个类中,因此您的模型似乎正在尝试将随机输入分类到您的自定义类中。如果您可以提供有关自定义类的更多信息,将很有用。

我想您可以尝试在班级预测中引入信心成分,即,假设您使用softmax,对于任何输入,只有当班级给出的概率高于某个阈值时,您才可以真正接受分类,否则分类为随机。

此外,您可以尝试添加另一个自定义类,并使用所有随机符号来训练模型,但我认为这不是理想的选择。