我想知道如何使一个自动标记的图像。
我尝试过使用Tensorflow并对模型进行了多次预训练。首先,它对分类非常好。
但现在,我需要做自动标记。
使用TensorFlow,预测和结果将始终为1。
例如,类似于以下内容:
xxx.jpg预测结果:
- Cat = 0.822
- 犬= 0.177
- 鹿= 0.001
总和总是1。
我想要的是这样的东西:
xxx.jpg预测结果:
- 类别= 0.901
- 犬= 0.811
- 鹿= 0.991
因为在xxx.jpg
的同一张图片中可能有猫、狗和鹿(就像Clarifai一样)。
我想知道实现这一目标的基本概念是什么?
谢谢大家。
1条答案
按热度按时间qyswt5oh1#
看看你做的最后一层。正如你所说的,你的预测之和总是1,这听起来像你应用了softmax(https://en.wikipedia.org/wiki/Softmax_function)。如果你删除它,你会得到每个对象的激活。
请让我知道这是否对你有帮助!