Home > Blockchain >  Convert Tensorflow model with multiple inputs to OpenVino IR format
Convert Tensorflow model with multiple inputs to OpenVino IR format

Time:05-21

When trying to convert a custom model built using tensorflow I came across the following issue. The model has two inputs and is currently in the saved_model format. When running:

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

An error is thrown

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

How am I supposed to provide the input names?

CodePudding user response:

When specifying input shapes for several layers, you must provide names for the inputs for which shapes will be overwritten. For multiple input topology, specify the inputs as below:

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

Refer to Converting a TensorFlow Model documentation.

  • Related