在 Windows 上從 .pb 轉換為 .tflite 時出錯

Question

我有一個凍結的 tensorflow 對象檢測模型frozen_inference_graph.pb ，我需要將它轉換為.tflite格式以在 android 應用程序中使用它。

我按照這里的建議使用tflite_convert ， https: tflite_convert

tflite_convert 
--graph_def_file=frozen_inference_graph.pb 
--output_file=new_graph.tflite 
--input_format=TENSORFLOW_GRAPHDEF 
--output_format=TFLITE 
--input_shape=1,224,224,3 
--input_array=image_tensor 
--output_array=detection_boxes,detection_scores,detection_classes,num_detections 
--inference_type=FLOAT 
--input_data_type=FLOAT

這是我得到的錯誤：

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

編輯：我讀了tflite文檔和它被提及，僅使用凍結款freeze.py可以使用轉換tflite_convert 。 但我使用export_inference_graph.py來獲取frozen_inference_graph.pb 。 有沒有其他方法可以將對象檢測模型轉換為tflite 。 我正在使用模型ssd_mobilenet_v1_coco_11_06_2017 。

Answer 1

轉換器可以更改輸入和/或輸出類型，但只有一方面：如果模型具有浮點輸入張量，則可以創建具有quantized (uint8) input的 tflite 模型，但反之則不然。

從您的命令中刪除--input_data_type=FLOAT ，它應該可以工作。