android上部署tensorflow

https://www.jianshu.com/p/ddeb0400452f 按照这个博客就可以

https://github.com/CrystalChen1017/TSFOnAndroid 这个博客对应有一个github的仓库,里面有一个完整的demo,并且有编译好的jar和so文件

如果自己去编译jar so文件,实在是太麻烦,并且我尝试了一周都没有成功,还是直接使用这个博客的jar和so

需要自己生成一个pb文件,https://github.com/yscacaca/DeepSense/tree/master/android_test  deepsense的android_testpb下有生成pb的两个代码,照着跑就能生成,其实pb文件就是既能保存模型的图,也能保存模型的权重

 

因为模型本身使用了gru操作,所以生成的pbtxt中会有“All”这个op,但android上没有,就只能把整个gru部分剔除掉进行部署

No OpKernel was registered to support Op 'All' with these attrs

 

posted @ 2018-05-03 18:29  有梦就要去实现他  阅读(461)  评论(0编辑  收藏  举报