Tensorflow中正确图像分类的可能性

Question

我正在使用Tensorflow重新训练模型进行图像分类。 我正在做单标签分类。

我想为正确的分类设置一个阈值。 换句话说，如果最高概率小于给定阈值，则可以说图像是“未知”，即，如果np.max（results）<0.5->将标签设置为“未知”。

因此，是否有任何行业标准来设置此阈值。 我可以设置一个随机值，例如60％，但是是否有任何文献支持该阈值？

任何链接或参考将非常有帮助。

非常感谢。

Answer 1

神经网络不能“现成”地进行单标签分类。

您如何训练它？ 仅包含与您的目标域相关的数据？ 您的模型只会学习输出一个。

您有两种策略：

1. 您使用与“ HotDog或Not HotDog应用程序”相同的策略，则将整个imagenet放在两个不同的文件夹中，一个包含所需的类，另一个包含其他内容。

2. 您使用convnet作为特征提取器，然后使用第二种模型，例如One-Class SVM。

您必须了解，进行一类分类不是像二进制分类那样的简单直接的问题。