我已经使用ssd_mobilenet_v1_coco_2017_11_17模型训练了对象检测接口来检测自定义对象。但在训练之后,API只检测自定义对象,而不检测API已经训练过的对象。ssd_mobilenet_v1_coco_2017_11_17模型检测到90个对象。
ssd_mobilenet_v1_coco_2017_11_17
有没有办法将更多的类添加到现有的模型中,以便它可以检测到新的对象以及它已经训练过的对象?
发布于 2018-01-24 19:16:13
这个问题已经被问到了here,一些答案可以在here找到。
网络的最后一层是softmax层。当训练网络时,网络的权重针对训练集上的确切类别数进行优化。因此,如果你需要添加一个新的类以及它所训练的类,最简单的方法是获得它所训练的原始数据集以及你的新类图像。然后从预先训练好的模型权重开始训练。训练应该收敛得更快,因为它需要做相对较少的调整。
https://stackoverflow.com/questions/48419452
相似问题