替代Keras的Lambda层

Question

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

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

问题是，CoreML文件格式不支持Lambda图层。 我不是这个领域的专家，但据我所知，此处的Lambda图层用于使用ctc_batch_cost()计算损失。 该层围绕线464创建。 我想这比“内置”损失函数更精确。

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

Answer 1

成本函数通常不包含在CoreML模型中，因为CoreML仅进行推理，而成本函数用于培训。因此，在导出模型之前去除该层，您应该很高兴。