機器學習模型中有自定義layer時，無法透過TFLiteConverter把訓練好的模型轉成tflite

機器學習 keras tensorflow tensorflowlite

dietaler 2022-10-30 19:22:12 ‧ 586 瀏覽

小弟有個問題想要詢問。我的模型當中有自己定義的layer，而我想要把訓練好的模型從keras(.h5)轉成tflite以便之後在手機run，但是我去call官方給出的medthod:

tf.lite.TFLiteConverter.from_keras_model(model)

程式會說因為有未知的layer所以無法轉換成功，我想詢問是否有辦法解決呢??

直播研討會

{{ item.channelVendor }} {{ item.webinarstarted }} |

直播中

增廣建文

iT邦研究生 5 級 ‧ 2022-11-06 01:07:56

你的custom layer有用ops的形式定義嗎?
如果不符合tf能吃的格式就會壞掉

https://stackoverflow.com/questions/57865705/tflite-converter-from-a-custom-keras-layer

立即登入回答

參賽組數

1064 組

團體組數

40 組

累計文章數

22189 篇

完賽人數

600 人

IT邦幫忙