1.模型轉(zhuǎn)換
將pb模型轉(zhuǎn)換成tflite模型,注意op的兼容性水慨。當時錯誤在tf.cast不支持。
python上驗證轉(zhuǎn)換后的tflite模型。
2.android上部署
官方demo的aar獲取不到
implementation 'org.tensorflow:tensorflow-lite:0.0.0-nightly
替換成:
implementation 'org.tensorflow:tensorflow-lite:+
3.Gpu部署
需要在同一個線程中使用棺克。