首页 > 解决方案 > AutoML VISION Google SingleLabel 分类输出 TopK 结果

问题描述

目前AutoML Vision API正在输出带有相应分数的 SingleLabel

例如:

我用 3 个类训练了模型:

  1. 一个
  2. C

然后,当我使用Test & Use并上传另一张图片时,我只得到 [CURRENT OUTPUT]

  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-platformvisiongoogle-cloud-automlautoml

解决方案


单标签分类为每个分类图像分配一个标签,它只返回一个预测类。

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

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

在此处输入图像描述

如果出于任何原因,您希望可以选择在单标签分类上获得所有/k 个预测类别的分数,我建议您提出Feature Request


推荐阅读