多标签图像的最快方法?

数据挖掘 图像分类 计算机视觉 标签
2022-03-08 20:21:53

我想制作一个包含数千个(不同大小)图像的新数据集。现在我需要为每个图像分配多个标签。

当然,我已经看过 github 等,那里有很好的标签程序。问题是您似乎总是只能为图像分配一个标签。

您知道为图像分配多个标签并将它们保存在专用文件中的简单方法吗?

1个回答

您的问题相对缺乏细节,所以我真正建议的是使用标准的卷积网络分类器。

您似乎遇到的问题是,大多数此类模型输出一个单热向量,即分类分布,对应于单个标签:,其中(并且只有标签分配给输入的概率。通常这是通过为某些输入图像和神经网络计算的。p=(p1,,pn)ipi=1piiip=SoftMax(fθ(x))xfθ

要更改它以处理多个标签,让,其中是将标签分配给输入的概率。您可以将每个视为对单个标签上的伯努利分布进行参数化。例如,您可以,其中是元素 sigmoid 函数。然后,您可以通过例如对每个分配标签,即,如果分配给,其中q=(q1,,qm)qi[0,1]iqqq=σE(fθ(x))σExqjkxxkTT是一个超参数,可以是 0.5。

换句话说,使用 Github 上的分类器,将 softmax 替换为 elementwise sigmoid,将更改为,并使用您的训练数据;就是这样。nm

希望这就是你的意思。