AutoML VISION Google SingleLabel 分类输出 TopK 结果

Posted 2023-03-28

【中文标题】AutoML VISION Google SingleLabel 分类输出 TopK 结果

【英文标题】：AutoML VISION Google SingleLabel Classification output TopK results

【发布时间】：2021-06-27 16:59:50

【问题描述】：

目前 AutoML Vision API 正在输出具有相应 Score

的 SingleLabel

例如：

我用 3 个类训练了模型：

一个 B C

然后，当我使用 Test & Use 并上传另一张图片时，我只得到 [当前输出]

Class A and 0.988437 / 0.99

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

Class A and 0.988437 / 0.99 Class C and 0.3551 / 0.36 Class B and 0.1201 / 0.12

根据他们的分数排序。

提前致谢。

【参考方案1】：

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

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

在 UI（您似乎正在使用的界面）中，您可以在 create your dataset 时指定您希望自定义模型执行的分类类型。

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

【讨论】：

我使用他们的 API 和他们提供的 JSON 解决了​​这个问题。但是如果他们能添加这个功能那就太好了。另外，他们是如何进行 AutoML 培训的？是否有所有可用的技术细节？我在 GitHub 上看到了 NAS，但没有别的

如果您引用 automl/NASLib，这不是 Google 的官方 repo。并且 AFAIK 的技术细节不可用。 AutoML 就是为没有这种专业知识的人提供 ML 能力。

关于您的解决方案，您能否分享它或分享您遵循的文档，因为当我尝试 API 时，预测的类也是一个（对于每个图像的单标签分类）。这将有助于任何未来遇到同样问题的用户。