如何为图像分类器提供默认输出?

How to have a default output for an Image Classifier?

我使用 CNN 编写了一个简单的图像分类器,将输入图像分类为标签:狗或猫。 当显示 Cats/Dogs 以外的图像时,CNN 将其分类为任何一个标签(狗或猫),这是不正确的。 我希望我的 CNN classify/label 所有其他随机图像(房子、人、鸟等)作为无效输入。如何做到这一点?

有两种方法可以实现..

1 . 在你的训练数据中包含第三个 class 具有任何随机图像,如房子、人、鸟等,这样在推理时你会得到这个随机数据预测(如果包含所有可能的最大图像量)。

2 . 在你的 softmax class 之后使用阈值如果任何低于 0.7 阈值的东西(在做一些试验后根据你的意愿改变)将是未知 class(这个技巧我经常用)