Tensorflow Inception v3 retraining -将文本/标签附加到单个图像

Question

我正在使用inception v3模型来重新训练我自己的数据集。我有一些代表类的文件夹，其中包含每个类的图像。我想做的是将一些文本ids‘附加’到这些图像上，这样当它们被重新训练并用于运行分类/相似性检测时，这些ids也会被检索到。(基本上是它的图像相似性检测)

例如，图像X属于“教师”类，它属于John。当我重新训练模型，并对新模型运行分类时，我希望获得教师类，但除此之外，我还希望知道谁是教师(John)。

你知道该怎么做吗？

问候

Answer 1

你有3个主要的选择--增加你的班级，多标签学习或者训练几个模型。

第一个选项是最直接的--你可以让班级是Teachers_John的老师和班级是Teachers_John的班级的老师，而不是属于John的老师和属于Jane的老师，并学习像学习任何其他类别集一样对这些类别进行分类，或者使用类似hierarchical softmax的东西。

第二个选项是有一组类别，包括教师以及约翰和简-现在你的目标不是正确预测一个最准确的类别(教师)，而是几个类别(教师和约翰)。

最后一个选项是创建一个模型层次结构，其中第一个模型学习区分John和Jane，其他模型学习为每个模型的内部类进行分类。