tensorflow lite部署

2020-01-07  本文已影响0人  小王子威威

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部署

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

上一篇 下一篇

猜你喜欢

热点阅读