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

Question

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

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

Answer 1

此问题已经提出here，可以找到一些答案here。

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