無法將使用 Docker 創建的凍結 Inception graph.pb 轉換為 .tflite

Question

我在我的 macOS 設備中通過 Docker 創建了我的saved_model.pb 。

嘗試tflite_convert --output_file=./myModels/mymodel.tflite --saved_model_dir=./myModels/后，它輸出以下錯誤，需要我一個標簽參數：

“ValueError：如果存在多個 MetaGraph，則使用 tf.saved_model.load 導入 SavedModel 需要 'tags=' 參數。得到 'tags=None'，但 SavedModel 中有 0 個帶有標簽集 [] 的 MetaGraph。傳遞一個'tags=' 參數來加載這個 SavedModel。”

我還嘗試了另一個命令，您必須提供輸入和 output arrays 但我知道哪些是我的 arrays 放入這些字段。 有人遇到過這個問題並解決過嗎？ 謝謝。

Answer 1

您可以使用 saved_model_cli 在您保存的 model 中找到標簽：

https://www.tensorflow.org/guide/saved_model#saved_model_cli

$ saved_model_cli show --dir./myModels/ --all

使用--saved_model_tag_set將所需的標簽傳遞給 tflite_convert。