在AutoML Vision中进行预测时如何包含-other-标签

时间:2018-12-05 08:10:01

标签: google-cloud-automl

我已经训练了带有3个标签的模型,但是在将其用于预测阶段时,我上传了不属于这3个标签的图像,但它仍然检测到该图像是3个标签之一。如何使返回的图像不是全部3个标签。如果使用第4个标签“ other”来训练模型,我不知道如何为该标签准备数据集。 有人有建议吗?感谢您的阅读,对不起我的英语不好

1 个答案:

答案 0 :(得分:0)

AutoML支持两种类型的模型:

  1. 多类别-这是默认类别-多类别分类假设每个样本都分配给一个标签,并且仅分配一个标签。请记住,它假定总是一个标签正确,并且会尽力始终返回至少一个标签(实际上,它会尝试确保所有预测得分的总和== 1)。

  2. 多标签-此模式不做任何假设,对于每张图像,可以检测到任意数量的标签-包括0。

您有两个选择:

  1. 创建数据集时,选中“启用多标签分类”复选框
  2. 添加其他类“ other”或“ None_of_the_above”,并将一些示例图像添加到此类。

选项1和2可以组合。