tensorflow - 使用 tensorflow 进行自定义对象检测

Question

我已经使用ssd_mobilenet_v1_coco_2017_11_17模型训练了对象检测 API 来检测自定义对象。但是在训练之后，API 只检测到自定义对象，而不是 API 已经训练过的对象。ssd_mobilenet_v1_coco_2017_11_17模型检测 90 个物体。

有没有办法向现有模型添加更多类，以便它可以检测新对象以及已训练的对象？

Answer 1

这个问题已经在这里问过了，一些答案可以在这里找到。

网络的最后一层是 softmax 层。当网络被训练时，网络的权重会针对训练集上的确切类数进行优化。因此，如果您需要添加一个新类以及它所训练的类，最简单的方法是获取它所训练的原始数据集以及您的新类图像。然后从预训练的模型权重开始训练。训练应该更快地收敛，因为它必须做相对较少的调整。