我已经使用 tensorflow 对象检测/SSD mobilenet 训练了一个模型。它工作得很好!
我想给它添加一个类 - 只是为了检测笔或其他东西。
我怎样才能做到这一点?
我已经创建了我的图像集,我只是找不到关于如何将单个类添加到现有模型的任何教程或信息。
谢谢!
最佳答案
您将类添加到现有模型的想法,用 tensorflow 对象检测 api 行话来说,就是在自定义数据集上重新训练自定义对象检测模型 (在这种情况下,您的笔数据集)。
关于如何使用 tensorflow 对象检测 api 构建自定义对象检测器,有很多很好的教程。
比如senddex贴了一个很好的step by step教程here .官方 github repo 页面也包含一些像这样的好教程:bringing in your own dataset ,从某种意义上说,这实际上与从预训练模型中添加或删除类相同。
但同样,我认为上述教程并没有达到 的确切目标。添加类(class) 对于模型,如果您有旧类和新类的数据并重新训练所有这些类,它只会添加新类。由于在您的情况下,您只有新类的数据,因此更正式地将其称为重新训练自定义对象检测模型。
关于tensorflow - 如何将类添加到现有模型?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56594290/