如何将tensorflow-lite模型转换为coreml .mlmodel

Question

我当前正在使用与我的ios应用集成的google firebase。我在我的应用程序中将MLKit与本地TensorFlow lite模型用作图像分类器。

但是有了Xcode 9，我无法使用我的应用程序执行此操作，因为autoMLVision是新的，并且需要xcode 10.1。现在我认为最好的方法是将我在google firebase中训练并下载的TensorFlow lite模型转换为.mlmodel并使用Core ML进行图像分类。

我将如何进行此转换。如果您有其他解决此版本问题的方法，请分享。告诉我是否需要更多信息。

Answer 1

据我所知，您目前无法将TFLite模型直接转换为Core ML。您只能使用tfcoreml转换冻结的.pb模型。

一个选项可能是将TFLite模型转换回冻结的.pb模型（不确定是否有转换工具可以使您执行此操作）。

另一种选择是使用coremltools的NeuralNetworkBuilder手动进行转换。您将需要阅读TFLite文件，提取权重，将其潜在地转换为Core ML期望的顺序，然后使用NeuralNetworkBuilder构建模型的Core ML版本。