将具有多个输入的 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 文档。
尝试转换使用 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 文档。