将 Google 的 SavedModel 转换为 Apple 的 mlmodel

Convert Google's SavedModel to Apple's mlmodel

本周 Apple announced 支持经过训练的 ML 模型。

如何将经过训练的 Tensorflow 模型(Google SavedModel 格式的云机器学习引擎模型)转换为 Apple Core ML 模型(.mlmodel 格式)?

您应该使用 coremltools(python 包)。

查看 WWDC Session 710 Video "Core ML In Depth"

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