使用 CoreMLTools 为 iOS 11/12 生成模型

Question

我们有一个使用 Keras 训练的模型，使用 MobileNetV2 架构。

我们可以使用 CoreMLTools 将 .H5 文件转换为 .MLModel CoreML 模型。

但是，使用最新的 CoreMLTools (5.x) 生成的模型只能在 iOS 13 及更高版本上运行，但我们的应用支持 iOS 11。

有没有办法使用最新的 CoreMLTools 生成兼容 iOS 11/12 的模型？

我们曾考虑尝试安装较旧的 CoreMLTools（如 2.x），但由于其他原因在安装时遇到了依赖性问题。但是感觉应该有办法在转换模型的时候指定CoreML版本？

Answer 1

我会强烈建议解决您的版本依赖关系并让旧版本的 coremltools 正常工作。我理解那里的困难，但我向你保证，所有其他道路都会更加困难。

现在好消息。 CoreML 模型只是协议缓冲区，您可以在没有 coremltools 的情况下轻松加载和操作自己。我将他们的协议规范的编译版本保存在一个库中，仅用于这些类型的任务。您可以在此处获取 PB 规范：https://github.com/apple/coremltools/tree/f19052c7f113740069bfac7b0291c5c6c9571ca6/mlmodel/format

在 PB 查看器中加载您的 iOS 11 模型，加载 iOS 13 版本，然后删除 13 版本中所有不在 11 中的内容。:-)

谢天谢地，CoreML 模型非常简单，我猜你只需要重置一个版本标志集。