TensorFlow遇见Android Studio 构建App入门教程

一、PC端搭建模型与训练，保存模型为PB文件

'''

简单起见，创建一个最简单的tensorflow模型，没有实际功能

只为了演示在android studio搭建App

运行代码就会在 你选择的本地目录找打PB文件

'''

import tensorflow as tf

 

# 此处的输入层定义为常量，为了简单起见，输入层名字为input，简单起见

input_constant = tf.constant([1,2],dtype=tf.float32,name="input")

 

#输出层名字为output，简单起见

out_data = tf.add(input_constant,input_constant,name="output")

 

# Session运行，这样就有了模型参数

sess = tf.Session()

sess.run(tf.global_variables_initializer())

print(sess.run(out_data))

 

#把当前的模型保存为PB文件,PB文件会保存当前tensorflow的模型,将其他值固化为常量

# 第一个参数 sess指定为当前的Session

# 第二个参数是要保存的 图的定义,默认是当前图

# 然后是要输出的节点

output_graph_def = tf.graph_util.convert_variables_to_constants(sess, sess.graph_def, output_node_names=['output'])