Tensorflow自动标记

lf5gs5x2  于 2023-03-30  发布在  其他
关注(0)|答案(1)|浏览(90)

我想知道如何使一个自动标记的图像。
我尝试过使用Tensorflow并对模型进行了多次预训练。首先,它对分类非常好。
但现在,我需要做自动标记。
使用TensorFlow,预测和结果将始终为1。
例如,类似于以下内容:
xxx.jpg预测结果:

  • Cat = 0.822
  • 犬= 0.177
  • 鹿= 0.001

总和总是1。
我想要的是这样的东西:
xxx.jpg预测结果:

  • 类别= 0.901
  • 犬= 0.811
  • 鹿= 0.991

因为在xxx.jpg的同一张图片中可能有猫、狗和鹿(就像Clarifai一样)。
我想知道实现这一目标的基本概念是什么?
谢谢大家。

qyswt5oh

qyswt5oh1#

看看你做的最后一层。正如你所说的,你的预测之和总是1,这听起来像你应用了softmax(https://en.wikipedia.org/wiki/Softmax_function)。如果你删除它,你会得到每个对象的激活。
请让我知道这是否对你有帮助!

相关问题