使用Keras 2保存CoreML负载模型

时间:2017-06-15 19:54:49

标签: python keras ios11 coreml

Apple的新CoreML可以与受欢迎框架培训的模型配合使用。至少他们在18岁和14岁时这样说。在WWDC video。但是在docs中似乎关于神经网络它们只支持Caffe和Keras 1.2.2(see code),而Keras它支持2.0版本,而Tensorflow和Theano在他们自己的权利中非常受欢迎。

要使用Keras 2运行转换,有没有比自己实现转换更好的方法? coremltools包中的Keras转换代码长约2000行,我不太了解所有Keras模型表示,所以我真的不想走这条路。

我已尝试直接转换使用Keras 2保存的模型,但这不起作用并且失败

TypeError: ('Keyword argument not understood:', u'gamma_initializer')

来自site-packages/keras/engine/topology.py", line 326

他们也没有针对coremltools的github,因此很难更详细地讨论这个软件包。我已上传了该软件包的副本:https://github.com/gsabran/coremltools

1 个答案:

答案 0 :(得分:3)

Keras 2.0支持已经存在(昨天发布),coremltools v.0.4.0

另请参阅Apple Developer Forums

上的最新评论