Keras 中 Lambda 层的替代方案

Question

我尝试将Keras OCR example 转换为CoreML 模型。

我已经可以训练我稍微修改过的模型，并且在 Python 中一切看起来都不错。但现在我想将模型转换为 CoreML 以在我的 iOS 应用程序中使用它。

问题是，CoreML 文件格式不支持 Lambda 层。 我不是这个领域的专家，但据我了解，这里的 Lambda 层是用来计算损失的，使用ctc_batch_cost()。 该层围绕第 464 行创建。 我想这是为了比“内置”损失函数更精确。

有没有什么方法可以重写模型创建以适应 CoreML 支持的层集？ 我不知道该模型使用哪种输出层类型。

Answer 1

CoreML 模型中通常不包含成本函数，因为 CoreML 只进行推理，而成本函数用于训练。所以在你导出模型之前去掉那个层，你应该很高兴。