请教个问题,比如图像分类中,我要分猫和狗两个类别,训练好模型后传入其他动物图像如老鼠,后也会根据概率被分为猫狗中,有什么好的办法去分类单类别分类吗,比如来了一张图像,如何判断他不属于猫狗类,而不是属于猫狗的哪一类,根据最后一层的softmax判断不太准确,有时候一张不相关的图像概率也会很高,有什么传统机器学习方法或者深度学习方法吗?
你在训练模型的时候只有猫狗图片和类别,学到的模型自然只能区分这两种,你输入张老鼠他也只会给到这两个其中一种。你可以在训练的时候再加一个类别,或者你先做个猫狗图和其他类型图片的分类器。如果你只想在改下原有的模型,可以更改下最后一层,简单的改为nn.linear(),最后得到两个值,然后用网格的方法找到一个f1最高的阈值,从而判断属于哪个类,低于某个值就是其他类型。
[cat,dog,others]