无法将使用Docker创建的冻结的Inception图.pb转换为.tflite

Question

我已经在我的macOS设备中通过Docker创建了一个saved_model.pb。

尝试tflite_convert --output_file=./myModels/mymodel.tflite --saved_model_dir=./myModels/后，它输出以下错误，要求我使用tag参数：

  

“ ValueError：如果存在多个MetaGraph，则导入带有tf.saved_model.load的SavedModel需要一个'tags ='参数。得到了'tags = None'，但是带有标签集[]的SavedModel中有0个MetaGraphs。 。传递一个'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。