无法将使用 Docker 创建的冻结初始图 .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。"

我还尝试了另一个命令，您必须在其中提供输入和输出数组，但我知道哪些是我要放入这些字段的数组。有人遇到过这个问题并解决过吗？谢谢。

Answer 1

您可以使用 saved_model_cli 在您保存的模型中找到标签：

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

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

传递所需的标签 --saved_model_tag_set 到 tflite_convert。