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

Question

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

但是有了Xcode9，我无法用我的应用程序做到这一点，因为autoMLVision是新的，需要Xcode10.1。现在，我认为最好的做法是将我在google firebase中训练并下载的Core模型转换为.mlmodel，并使用TensorFlow ML进行图像分类。

我该如何进行这个转换呢？如果您有解决此版本问题的替代方法，请分享。如果您需要更多信息，请告诉我。

Answer 1

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

一种选择可能是将TFLite模型转换回冻结的.pb模型(不确定是否有转换工具可以这样做)。

另一种选择是手动进行转换，使用coremltools的NeuralNetworkBuilder。您将需要读取TFLite文件，提取权重，可能会将它们转置为Core期望的顺序，然后使用NeuralNetworkBuilder构建模型的Core版本。