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