Tensorflow 使用图形文件的Tf到tflite转换问题
**Tf到tflite的转换问题 我运行的是tensorflow版本1.13,我提供了tf lite所需的图形文件路径、输入数组和输出数组、输入形状**Tensorflow 使用图形文件的Tf到tflite转换问题,tensorflow,tensorflow-lite,Tensorflow,Tensorflow Lite,**Tf到tflite的转换问题 我运行的是tensorflow版本1.13,我提供了tf lite所需的图形文件路径、输入数组和输出数组、输入形状** input_arrays=['DecodeJpeg/contents']#Input tensor output_arrays=['final_result']#output tensor converter=tf.contrib.lite.TFLiteConverter.from_frozen_graph(graph_def_file
input_arrays=['DecodeJpeg/contents']#Input tensor
output_arrays=['final_result']#output tensor
converter=tf.contrib.lite.TFLiteConverter.from_frozen_graph(graph_def_file,input_arrays,output_arrays,input_shapes={'DecodeJpeg/contents':()})
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)
这里converter.convert()将我抛出下面的错误
if not shape or not shape.as_list():
raise ValueError("Provide an input shape for input array "
"'{0}'.".format(_tensor_name(tensor)))
shape_list = shape.as_list()
if None in shape_list[1:]:
ValueError:为输入数组“DecodeJpeg/contents”提供输入形状。在代码中,必须为输入数组指定输入形状。
请将
input_-shapes={'DecodeJpeg/contents':()}
更改为input_-shapes={'DecodeJpeg/contents':None}
或根据您的输入形状如下input_-shapes={'DecodeJpeg/contents':[1,16,16,3]}
在您的代码中,必须为您的输入数组指定输入形状。
请将
input_-shapes={'DecodeJpeg/contents':()}
更改为input_-shapes={'DecodeJpeg/contents':None}
或根据您的输入形状如下input_-shapes={'DecodeJpeg/contents':[1,16,3]}
我的输入形状是一个标量。所以我只是用一个tuple@sravankumar,您是否已尝试分配输入形状或仅尝试输入形状=无。我的输入形状是标量。所以我只是用一个tuple@sravankumar,您是否已尝试分配输入形状或仅尝试输入形状=无。