tensorflow lite部署

1.模型转换

将pb模型转换成tflite模型,注意op的兼容性。当时错误在tf.cast不支持。

python上验证转换后的tflite模型。

2.android上部署

官方demo的aar获取不到

implementation 'org.tensorflow:tensorflow-lite:0.0.0-nightly

替换成:

implementation 'org.tensorflow:tensorflow-lite:+

3.Gpu部署

需要在同一个线程中使用。

你可能感兴趣的:(tensorflow lite部署)