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

Question

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

ImportError：无法导入名称“relu6”

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

完整的 colab 文件在这里：

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

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

标签： tensorflowkerascoremlcoremltools

Answer 1

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

写信keras.__version__看看你的笔记本真正使用的是什么版本的 Keras。

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