本周,Apple宣布支持经过训练的 ML 模型。
如何将经过训练的 Tensorflow 模型(SavedModel 格式的 Google Cloud 机器学习引擎模型)转换为 Apple Core ML模型(.mlmodel 格式)?
这取决于您的 TensorFlow 模型包含什么。coremltools 不支持 TensorFlow,因此您必须编写自己的转换器。但这仅在您的模型仅包含 mlmodel 格式支持的内容时才有效。您可以从 coremltools 网页下载此格式的规范。由于可以在 TensorFlow 中构建任意复杂度的计算图,因此 coremltools 目前不支持它也就不足为奇了(因为 mlmodel 本身只支持有限数量的模型类型)。
您应该使用 coremltools(python 包)。