如何将类添加到现有模型？

Question

我已经使用张量流对象检测/ SSD移动网络训练了一个模型。效果很好！

我想在其中添加一个类-只是为了检测笔或其他东西。

我该怎么做？

我已经创建了我的图像集，但是我找不到关于如何向现有模型添加单个类的任何教程或信息。

谢谢！

Answer 1

您要在现有模型中添加类的想法，使用tensorflow对象检测API术语来说是要在自定义数据集上重新训练自定义对象检测模型（在这种情况下，是笔数据集）

关于如何使用tensorflow对象检测api构建自定义对象检测器，有很多不错的教程。

例如，sentdex发布了一个很好的循序渐进教程here。官方的github回购页面也包含一些类似这样的好教程：bringing in your own dataset，从某种意义上说，这实际上与从预训练的模型中添加或删除类相同。

但是再次，我认为上述教程并没有达到将模型添加类的确切目标，只有在您拥有旧类和新类的数据并重新培训的情况下，才添加新类在所有这些上。由于在您的情况下，您只有新类的数据，因此它在形式上通常称为重新训练自定义对象检测模型。

Answer 2

您可以使用现有的权重预训练权重，通过添加新类的数据来重新训练模型。