无法将使用 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。