Блог

** Проблема с преобразованием 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:]:
  

Ошибка значения: укажите форму ввода для входного массива ‘DecodeJpeg / contents’.

В вашем коде вы должны назначить форму ввода для вашего входного массива. Пожалуйста, измените input_shapes={'DecodeJpeg/contents':()} на input_shapes={'DecodeJpeg/contents':None} или в соответствии с вашей формой ввода, подобной этой input_shapes={'DecodeJpeg/contents':[1,16,16,3]} .