使用 Keras 2 保存的 CoreML 加载模型

CoreML load model saved with Keras 2

Apple 的新 CoreML 可以与在流行框架上训练的模型一起使用。至少他们在 WWDC video. But in the docs it seems that concerning neural nets they only support Caffe and Keras 1.2.2 (see code) 的 18" 上是这样说的),而 Keras it on it 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

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

另请参阅 Apple Developer Forums

上的最新评论