是否有用于Tensorflow对象检测的C / C ++ API

时间:2017-12-14 17:31:48

标签: tensorflow object-detection

是否有使用Imagenet数据集预先培训的C / C ++ API用于检测?

我曾尝试过Yolo ./darknet -i 0探测器演示cfg / imagenet1k.data extraction.cfg extract.weights

但它给了我错误 最后一层必须产生检测

对于Tensorflow,看起来只有python API https://github.com/tensorflow/models/tree/master/research/object_detection

1 个答案:

答案 0 :(得分:0)

当您在TensorFlow中开发模型时,它可以作为protobuf文件输出(通常带有pb扩展名,有关TensorFlow check out this page中protobuf的更多详细信息)。然后,可以将此protobuf文件用于以TensorFlow绑定的语言编写的不同应用程序中。 A simple tutorial on how to accomplish this for a C++ application can be found here.

关于Yolo,您可以从Yolo脚本生成protobuf文件,如下所示:

{{1}}

(有关可以传递给Yolo的其他参数的更多详细信息,请参见the Github readme page)。

然后可以将输出protobuf文件加载到C ++应用程序中以执行对象检测。