将具有多个输入的 Tensorflow 模型转换为 OpenVino IR 格式

Convert Tensorflow model with multiple inputs to OpenVino IR format

尝试转换使用 tensorflow 构建的自定义模型时,我遇到了以下问题。 该模型有两个输入,目前采用 saved_model 格式。 当运行:

python mo_tf.py --saved_model_dir C:/path_to/model_saved --input_shape [1,128,128,3],[1,1,1,64]

抛出错误

[ ERROR ] Please provide input layer names for input layer shapes.

我应该如何提供输入名称?

为多个图层指定输入形状时,您必须为要覆盖形状的输入提供名称。对于多输入拓扑,指定输入如下:

python mo.py --saved_model_dir --input input_name_1,input_name_2 --input_shape [input_shape_1],[input_shape_2]

请参阅 Converting a TensorFlow Model 文档。