我正在尝试使用批量图像在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);
我应该在程序中更改什么来获得具有批量大小的相应尺寸的输出?
提前谢谢!