Android Tensflow 训练模型移植 Android studio 学习

今天记录下tensflow 的学习内容

将 tensorflow模型移植到Android apk中,需要三个文件
1.python 训练模型生成的 .pb 文件
2.tensorflow的so文件libtensorflow_inference.so
3.tensorflow的jar包libandroid_tensorflow_inference_java.jar

大致过程如下
1.创建Android studio项目,创建过程与一般Android APK创建无区别
2.将.pb文件放到assets文件夹下(可能需要自己创建该目录)
3.将tensorflow的so文件及jar包放到libs下,并在需要通过lib文件(project struct设置中)加载到app项目中
4.代码中加载so文件,调用tensorflow jar函数使用该模型,让在app中运行起来

一、python 训练模型
1.搭建tensorflow环境请搜索下相关文章
2.生成 .pb 文件使用的是下面链接的代码,在此不再显示:
https://blog.csdn.net/cxq234843654/article/details/71171293

3.下面说下tensorflow下运行 .py 脚本注意项,其他的和平时使用python一致,如下面代码 hello.py :

 #!/usr/bin/env python
import tensorflow as tf
sess = tf.Session()
hello = tf.constant("hello World!",dtype=tf.string)
print(sess.run(hello))

1).脚本中一定要 写 #!/usr/bin/env python
2).有的在运行是要 $: python ./hello.py , 单单的 ./hello.py 也会报错

   import: unable to open X server

3).针对保存训练模型 .pb 文件时(不是上面这个例子)注意保存的路径 “./modem/xxx.pb” ,则对应路径下需要存在modem文件夹,否则直接 保存为 “./xxx.pb”即可,否则会报错

待续…

猜你喜欢

转载自blog.csdn.net/shiluohuashengmi/article/details/81583866