我正在使用初始v3模型重新训练我自己的数据集。我有几个文件夹代表包含每个类的图像的类。我想做的是附加'这些图像的一些文本ID,因此当它们被重新训练并用于运行分类/相似性检测时,也会检索这些ID。 (基本上是它的图像相似性检测)
例如,图像X属于班级'教师'它属于约翰。当我重新训练模型,并对新模型进行分类时,我想获得教师课程,但除此之外,我想知道谁是老师(约翰)。
任何想法如何去做?
此致
答案 0 :(得分:1)
您有3个主要选项 - 繁殖您的课程,多标签学习或培训多个模型。
第一个选项是最直接的选择 - 不是让属于约翰的老师和属于简的老师,你可以让老师的班级是Teachers_John,班级的班级是Teachers_John,并学会按照你的意愿分类到这些类别学习任何其他类别,或使用分层softmax。
第二种选择是拥有一组包括教师以及约翰和简的类别 - 现在你的目标不是正确预测一个最准确的班级(教师)而是几个(教师和约翰)。
你的最后一个选择是创建一个模型层次结构,其中第一个学习区分John和Jane以及其他人对每个模型的内部类进行分类。