替代Keras中的Lambda层

Question

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

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

问题是，CoreML文件格式不支持Lambda层。 我不是这方面的专家，但据我所知，这里的Lambda层用于使用ctc_batch_cost()来计算损失。 图层在线464周围创建。 我想这是用于“内置”损失函数更高的精度。

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

Answer 1

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