Tensorflow C ++ API批量推断

时间:2018-06-07 14:12:42

标签: tensorflow

我正在尝试使用批量图像在Tensorflow的C ++ API中对训练有素的SSD模型执行推理任务。看起来它有效,因为运行时间随着批量大小的增加而增加。但是我无法理解如何探索净输出,因为它保持相同的大小。

我将输入张量定义如下,batchSize是我的批量大小: Tensor input_tensor(tensorflow :: DT_UINT8,tensorflow :: TensorShape({batchSize,input_height,input_width,input_depth})); 输出张量是: std :: vector outputs;

会话运行操作: 状态run_status = session->运行({{input_layer,input_tensor}},output_layer,{}和& outputs);

我应该在程序中更改什么来获得具有批量大小的相应尺寸的输出?

提前谢谢!

0 个答案:

没有答案