Tensorflow自动标记

Question

我想知道如何为图片制作自动标记。

我尝试过tensorflow并多次训练模型。 首先，分类非常好。 但现在，我需要进行自动标记。

使用张量流，预测和结果将始终为1。

例如：

xxx.jpg预测结果： - Cat = 0.822 - 狗= 0.177 - 鹿= 0.001 总和将始终为1。

我想要的是这样的：

xxx.jpg预测结果： - Cat = 0.901 - 狗= 0.811 - 鹿= 0.991

因为在xxx.jpg的同一张照片中可能有猫，狗和鹿。 就像Clarifai（https://www.clarifai.com/）那样。

我想知道实现这个目标的基本概念是什么？

谢谢。

Answer 1

看一下你制作的最后一层。正如你所说，你的预测总和总是一个，这听起来像你应用softmax（https://en.wikipedia.org/wiki/Softmax_function）。 如果删除它，您将获得每个对象的激活。

请告诉我这是否对您有所帮助！