美文网首页
tensorflow lite部署

tensorflow lite部署

作者: 小王子威威 | 来源:发表于2020-01-07 09:44 被阅读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部署

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

相关文章

网友评论

      本文标题:tensorflow lite部署

      本文链接:https://www.haomeiwen.com/subject/wpdllctx.html