使用图形文件的TF到TFlite转换问题

Question

** 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,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'提供输入形状。

Answer 1

在代码中，您必须为输入数组分配输入形状。 请将input_shapes={'DecodeJpeg/contents':()}更改为input_shapes={'DecodeJpeg/contents':None}或根据您的输入形状，例如input_shapes={'DecodeJpeg/contents':[1,16,16,3]}。