图像分类，单类别分类，

Question

请教个问题，比如图像分类中，我要分猫和狗两个类别，训练好模型后传入其他动物图像如老鼠，后也会根据概率被分为猫狗中，有什么好的办法去分类单类别分类吗，比如来了一张图像，如何判断他不属于猫狗类，而不是属于猫狗的哪一类，根据最后一层的softmax判断不太准确，有时候一张不相关的图像概率也会很高，有什么传统机器学习方法或者深度学习方法吗？

Answer 1

你在训练模型的时候只有猫狗图片和类别，学到的模型自然只能区分这两种，你输入张老鼠他也只会给到这两个其中一种。你可以在训练的时候再加一个类别，或者你先做个猫狗图和其他类型图片的分类器。如果你只想在改下原有的模型，可以更改下最后一层，简单的改为nn.linear()，最后得到两个值，然后用网格的方法找到一个f1最高的阈值，从而判断属于哪个类，低于某个值就是其他类型。