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

我使用tensorflow对象检测/SSD mobilenet训练了一个模型。它工作得很好

我想添加一个类到它-只是为了检测钢笔或其他东西

我该怎么做

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

---

谢谢

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

用tensorflow对象检测api行话来说，将类添加到现有模型的想法是在自定义数据集上重新训练自定义对象检测模型（在本例中为笔数据集）

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

例如，sentdex发布了一个非常好的分步教程。另外，官方的github repo页面也包含了一些很好的教程，比如：从某种意义上讲，这实际上与从预训练模型中添加或删除类是一样的

---

但是，我认为上面的教程并没有达到将类添加到模型中的确切目标，它只是在您拥有旧类和新类的数据并对它们进行再培训的情况下添加新类。因为在您的案例中，只有新类的数据，所以更正式的说法是重新训练自定义对象检测模型。

将类添加到模型是一个研究问题，我们不知道如何做，所以这根本不是一个编程问题。@MatiasValdenegro我觉得TensorFlow有一种解决方法，你可以保存模型，用调整过的权重重新打开它，然后再培训？不完全一样，但似乎有一个轻微的解决方案哈哈！那么它可以检测预先训练的类+自己的类？