What are the parameters input_arrays and output_arrays that are needed to convert a frozen model '.pb' file to a '.tflite' file?

Question

I need to convert my .pb tensorflow model together with my .cpkt file to a tflite model to make it work in Mobile Devices. Is there any straight-forward way to find out how can I find what are the parameters I should use for input_arrays and output_arrays?

import tensorflow as tf

graph_def_file = "/path/to/Downloads/mobilenet_v1_1.0_224/frozen_graph.pb"
input_arrays = ["input"]
output_arrays = ["MobilenetV1/Predictions/Softmax"]

converter = tf.lite.TFLiteConverter.from_frozen_graph(
  graph_def_file, input_arrays, output_arrays)
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)

Shubham Panchal · Accepted Answer · 2019-03-24 02:32:27Z

2

According to the official docs here :

input_arrays: List of input tensors to freeze graph with.

output_arrays: List of output tensors to freeze graph with.

Meaning, input_arrays is the list of input tensors ( which are mostly placeholder tensors ). output_arrays is the list of Tensor objects which will act as outputs.

In your case, you are providing the name of the Tensor object. An actual Tensor object is required.

You can understand it with this example:

x1 = tf.placeholder( dtype=tf.float32 )
x2 = tf.placeholder( dtype=tf.float32 )
y = x1 + x2

input_arrays = [ x1 , x2 ]
output_arrays = [ y ]

You can learn to find the input and output tensors from here . Seeing your code, it seems that you know the tensor names, so you can refer this answer.

answered Mar 24, 2019 at 2:32

Shubham Panchal

4,3092 gold badges14 silver badges42 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Dani Over a year ago

So, you mean I must provide the tensors itself instead if the names? Thanks very much

Shubham Panchal Over a year ago

Yes. Provide the tensors and not their names. Also, accept the answer if it feels helpful.

Collectives™ on Stack Overflow

What are the parameters input_arrays and output_arrays that are needed to convert a frozen model '.pb' file to a '.tflite' file?

1 Answer 1

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related