替代Keras中的Lambda層

Question

我嘗試將Keras OCR example轉換爲CoreML模型。

我已經可以訓練我稍微修改過的模型，並且在Python中一切看起來都不錯。但是現在我想將模型轉換爲CoreML以使用它我的iOS應用程序。

問題是，CoreML文件格式不支持Lambda層。 我不是這方面的專家，但據我所知，這裏的Lambda層用於使用ctc_batch_cost()來計算損失。 圖層在線464周圍創建。 我想這是用於「內置」損失函數更高的精度。

有什麼辦法可以重新創建模型以適應CoreML支持的圖層集？ 我不知道該模型使用哪種輸出圖層類型。

Answer 1

成本函數通常不包含在CoreML模型中，因爲CoreML僅進行推理，而成本函數用於培訓。因此，在您導出模型之前剝離該圖層，您應該很好。