如何在TensorFlow对象检测API中合并训练后的类

时间:2018-10-29 16:02:49

标签: python tensorflow object-detection object-detection-api

TF OD API有一个标签图,您可以在其中告诉模型什么是类,并为其指定ID和名称。 假设我在给定的数据集上有一个训练有素的模型,例如MS COCO,我想在训练后合并/统一某些类(即无需任何微调)。 例如,我想将自行车和摩托车合并为“两轮车”类,将整个食物超类合并为单个类“ food”,并将整个动物超类合并为单个类“动物”。 是否可以通过修改标签映射,标签映射util或任何其他选项来使用当前的API?

一个人可以通过将相关类别的置信度添加到单个置信度中(假设置信度总和为1)以“复杂的方式”进行操作, 但是我可以将“ any”应用于具有最高置信度的类,即,如果具有最高置信度的类是合并集中的任何类,则将其分配给合并集中的类。

我尝试通过将要合并的类的名称更改为合并的集合类来修改标签图(例如,将“ bird”,“ cat”,“ dog”等命名为“ animal”),但是,在评估模型时,仍然会根据ID对ID进行不同的处理。 另一方面,我不能忽略要合并的ID,因为我仍然希望ID能够全部检测到它们,因此只能将它们视为相同。

谢谢。

0 个答案:

没有答案