训练Tensorflow仅一个对象

Question

相应的Tensorflow文档我训练了3个对象并获得结果（它可以识别这些对象）。当我显示其他对象（不是3个）时，它无法正常工作。 我只想训练一个物体（例如一个杯子），而只识别这个物体。是否可以通过Tensorflow进行操作？

Answer 1

您的问题没有提供足够的详细信息，但是我可以猜测您训练过的网络具有softmax激活和Categorical或SparseCategorical交叉熵损失。如果我的猜测是正确的，那么不管实际数据如何，这样的网络都会始终对三个类别之一进行预测，即没有“ no-one”选项。

为了训练网络仅识别一类对象，请仅使用一个通道和S型激活来进行唯一的输出。使用BinaryCrossEntropy损失来训练特定对象的模型。提供的数据集包括带有该对象的示例，没有该对象。