2017-04-19 71 views
1

我有一个tensorflow模型(再培训初始模型),可以分类5类车辆。现在我需要用这个训练好的模型为所有这5个类别制作一个物体探测器。可以通过删除最后一层来完成吗?任何人可以建议我如何进一步如何从图像分类器制作对象检测器?

回答

1

如果你真的需要使用你的训练有素的网络,那么你可以检测潜在的感兴趣的盒子,然后在每个网络上应用你的网络。这些框可以用“对象”方法来确定,例如EdgeBox。然而,在目前,物体检测通常是通过更为集成的方式获得的,例如通过faster RCNN获得的物体。这种方法集成了一个名为区域提议网络(RPN)的层,它可以确定感兴趣的区域,并与类别的识别相结合。

据我所知,最近最好的方法之一是Yolo,但它本来就是基于Darknet的。

+0

谢谢...我试过YOLO和SSD ...会尝试更快的RCNN ...我想我的想法类似于更快的RCNN –