将Google的SavedModel转换为Apple的mlmodel

Question

本周，Apple announced支持训练有素的ML模型。

如何将训练有素的Tensorflow模型（SavedModel格式的Google Cloud Machine学习引擎模型）转换为Apple Core ML模型（.mlmodel格式）？

Answer 1

这取决于您的TensorFlow模型包含的内容。 coremltools不支持TensorFlow，因此您必须编写自己的转换器。但是，只有当您的模型仅包含mlmodel格式支持的内容时，这才有效。您可以从coremltools网页下载此格式的规范。由于可以在TensorFlow中构建任意复杂度的计算图，因此coremltools目前不支持它并不奇怪（因为mlmodel本身仅支持有限数量的模型类型）。

Answer 2

你应该使用coremltools（python包）。

查看