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

时间:2018-12-24 12:28:23

标签: tensorflow keras coreml coremltools

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

  

ImportError:无法导入名称“ relu6”

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

完整的colab文件在这里:

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

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

1 个答案:

答案 0 :(得分:0)

您的安装似乎仍尝试使用Keras 2.2.0,因为coremltools中的错误是在检查Keras版本> = 2.2.0之后发生的。

编写keras.__version__来查看您的笔记本计算机实际使用的Keras版本。

请尝试安装较旧版本的keras_applications,该版本仍具有relu6功能。最近更改了。 Keras的问题在于,东西经常在次要版本之间来回移动。