如何将 class 添加到现有模型？
How to add class to existing model?
我已经使用 tensorflow 对象 detection/SSD mobilenet 训练了一个模型。效果很好！
我想给它添加一个 class - 只是为了检测笔或其他东西。
我该怎么做？
我已经创建了图像集，只是找不到任何关于如何将单个 class 添加到现有模型的教程或信息。
谢谢！
您可以使用现有的预训练权重通过为新 class 添加数据来重新训练模型。
您将 class 添加到现有模型的想法，用张量流对象检测 api 术语来说，是在自定义数据集上重新训练自定义对象检测模型（在本例中，您的笔数据集）。
有很多关于如何使用 tensorflow 对象检测构建自定义对象检测器的好教程 api。
例如senddex贴了一个很好的step by step tutorialhere. Also the official github repo page contains some good tutorials like this one: bringing in your own dataset，这实际上在某种意义上相当于在预训练模型中添加或删除classes
但是，我认为上述教程并没有达到 向模型添加 class 的确切目标，它只是添加新的 class 如果你有旧的 classes 和新的 classes 的数据并重新训练所有这些。由于在您的情况下您只有新 class 的数据，因此更正式地称为重新训练自定义对象检测模型。
我已经使用 tensorflow 对象 detection/SSD mobilenet 训练了一个模型。效果很好！
我想给它添加一个 class - 只是为了检测笔或其他东西。
我该怎么做？
我已经创建了图像集，只是找不到任何关于如何将单个 class 添加到现有模型的教程或信息。
谢谢！
您可以使用现有的预训练权重通过为新 class 添加数据来重新训练模型。
您将 class 添加到现有模型的想法，用张量流对象检测 api 术语来说，是在自定义数据集上重新训练自定义对象检测模型（在本例中，您的笔数据集）。
有很多关于如何使用 tensorflow 对象检测构建自定义对象检测器的好教程 api。
例如senddex贴了一个很好的step by step tutorialhere. Also the official github repo page contains some good tutorials like this one: bringing in your own dataset，这实际上在某种意义上相当于在预训练模型中添加或删除classes
但是，我认为上述教程并没有达到 向模型添加 class 的确切目标，它只是添加新的 class 如果你有旧的 classes 和新的 classes 的数据并重新训练所有这些。由于在您的情况下您只有新 class 的数据，因此更正式地称为重新训练自定义对象检测模型。