想把TensorFlow迁移到Android平台上,由于手机计算能力有限,因此需要对PB模型进行优化。

使用windows系统下自带的优化py文件,路径:

D:\Anaconda3\Lib\site-packages\tensorflow\python\tools\optimize_for_inference.py

在D:\Anaconda3\Lib\site-packages\tensorflow下打开命令行,将训练完成的retrained_graph.pb放到这里,然后输入:

python -m tensorflow.python.tools.optimize_for_inference \  --input=retrained_graph.pb \  --output=optimized_graph.pb \  --input_names="Mul" \  --output_names="final_result"  

可以看到已经正确生成相应的optimized_graph.pb优化后的文件。

再用这个生成后的pb进行测试,能够正确使用。

三 放入Android中测试

替换app\src\main\assets\model中对应的PB文件与txt文件,然后进行测试,成功加载PB模型。

更多相关文章

  1. 在Android中让Preference的宽度占满整个屏幕的宽度
  2. Android应用性能优化
  3. Android布局优化几个重要标签的使用include、merge、ViewStub、r
  4. [Android] ramdisk.img的生成及解压
  5. Android(安卓)网络框架Volley JAR包生成
  6. 如何在Android(安卓)NDK中调用第三方库文件(.so)
  7. Android代码实现WiFi连接
  8. android中生成keystore
  9. NDK HOWTO

随机推荐

  1. android中handler的使用
  2. [置顶] android 捕捉异常
  3. Android(安卓)封装http请求的工具类
  4. Android(安卓)打开闪光灯(手电筒)
  5. Android(安卓)多层树完美实现
  6. android 如何屏蔽Home键(长按短按都不起作
  7. Android(安卓)Telephony 分析[PART IV]
  8. android:Handler开启线程定时循环
  9. Android学习--Android带删除按钮的ListVi
  10. Linux下android ADT 安装不上的解决