无法将.pb文件转换为.tflite文件

Question

我正在尝试将save_model.pb转换为.tflite，但遇到错误。我正在运行的代码是

tflite_convert 
--output_file model.tflite 
--saved_model_dir   trained-inference-graphs/output_inference_graph_v1/saved_model 
--output_format TFLITE 
--inference_type FLOAT 
--input_arrays image_tensor 
--input_shapes 1,300,300,3 
--output_arrays detection_boxes,detection_classes,detection_scores,num_detections

我得到的错误是

F tensorflow/lite/toco/tooling_util.cc:2258] Check failed: array.data_type == array.final_data_type Array "image_tensor" has mis-matching actual and final data types (data_type=uint8, final_data_type=float).
Fatal Python error: Aborted

我正在使用Tensorflow版本1.14.0。

Answer 1

尝试

tflite_convert
--graph_def_file=tflite_graph.pb #your save model
--output_file=detect.tflite
--input_shapes=1,300,300,3 
--input_arrays=normalized_input_image_tensor
--output_arrays=TFLite_Detection_PostProcess,TFLite_Detection_PostProcess:1,TFLite_Detection_PostProcess:2,TFLite_Detection_PostProcess:3 
--allow_custom_ops