Sim*_*ely 5 object-detection neural-network tensorflow tfrecord
我使用张量流对象检测/SSD mobilenet 训练了一个模型。效果很好！
我想为其添加一个类 - 只是为了检测笔或其他东西。
我怎样才能做到这一点？
我已经创建了图像集，但我只是找不到有关如何将单个类添加到现有模型的任何教程或信息。
谢谢！
用张量流对象检测 API 术语来说，向现有模型添加类的想法是在自定义数据集（在本例中为您的笔数据集）上重新训练自定义对象检测模型。
关于如何使用 Tensorflow 对象检测 api 构建自定义对象检测器，有很多不错的教程。
例如，sentdex在这里发布了一个非常好的分步教程。另外，官方 github repo 页面包含一些很好的教程，例如：引入您自己的数据集，从某种意义上来说，这实际上与从预训练模型中添加或删除类相同。
但同样，我认为上述教程并没有达到向模型添加类的确切目标，只有当您有旧类和新类的数据并对所有类进行重新训练时，它才添加新类。由于在您的情况下，您只有新类的数据，因此更正式地称为重新训练自定义对象检测模型。