如何在 openvino 中扩展 blob 的 dims

Question

来自 OpenVINO 初学者的问候。在 official tutorial 中，为级联网络获取输入的最佳方式是

auto output = infer_request1.GetBlob(output_name);
infer_request2.SetBlob(input_name, output);

然而，在我的例子中，输出的布局是 CHW 但下一个网络的输入具有 NCHW 布局。那么我怎样才能有效地重塑或扩展 output 的暗淡呢？或者有没有更好的方法将 blob 提供给我的下一个模型？

我尝试了 input_info->setLayout(Layout::CHW);，它取自 openvino 的 hello_classification 示例，但它没有完成工作（我想我误解了这个功能）。

Answer 1

使用 InferenceEngine::CNNNetwork::reshape 为您的第一个没有批量维度的模型设置新的输入形状。