如何使用 coremltools 将 keras 模型转换为 corml 格式

Question

我想使用 coremltools 将我的 keras 模型转换为 coreml。 当我尝试这样做时，它给了我一个错误

ImportError: cannot import name 'relu6'

我的tensorflow版本是1.5.1 我的keras版本是2.1.6

完整的 colab 文件在这里：

https://colab.research.google.com/drive/1kSeErLsp_xaU37haUrwBO5jiNlV2RCll

我已经尝试过不同版本的模块，但我准备尝试一个我没有尝试过的新版本

Answer 1

看起来您的安装仍在尝试使用 Keras 2.2.0，因为 coremltools 中的错误发生在它检查 Keras 版本 >= 2.2.0 之后。

写 keras.__version__ 查看您的笔记本实际使用的 Keras 版本。

尝试安装旧版本的 keras_applications，它仍然具有 relu6 功能。它最近被改变了。 Keras 的问题是内容经常在次要版本之间移动。