AutoML VISION Google SingleLabel 分类输出 TopK 结果

问题描述 投票:0回答:1

当前 AutoML Vision API 正在输出带有相应 Score

的 SingleLabel

例如:

我用 3 个类别训练了模型:

  1. A
  2. B
  3. C

然后当我使用测试和使用并且上传另一张图像时,我只得到了 [电流输出]

  1. Class A and 0.988437 / 0.99

有没有办法可以通过 Top_K 类获得这种类型的输出(例如 Top 3 (k=3) ) [期望输出]

  1. Class A and 0.988437 / 0.99
  2. Class C and 0.3551 / 0.36
  3. Class B  and 0.1201 / 0.12

根据分数排序。

提前致谢。

google-cloud-platform computer-vision automl google-cloud-automl
1个回答
0
投票

单标签分类为每个分类图像分配一个标签,并且仅返回一个预测类别。

多标签更适合您的用例,因为它允许为图像分配多个标签。

在用户界面(您似乎正在使用的)中,您可以指定您希望自定义模型在创建数据集时执行的分类类型。

enter image description here

如果出于任何原因,您希望能够选择获取单标签分类的所有/k 个预测类别分数,我建议您提出功能请求

© www.soinside.com 2019 - 2024. All rights reserved.