A_Guide_to_Running_Tensorflow_Models_on_Android
A_Guide_to_Running_Tensorflow_Models_on_Android copied to clipboard
Optimized save method
This is not a issue,but a optimized save method.
- import dependency
from tensorflow.python.framework.graph_util import convert_variables_to_constants
- Convert variables to constants and save
output_graph_def = convert_variables_to_constants(sess, sess.graph_def, output_node_names=
['output/outputs'])
with tf.gfile.FastGFile('model/CTNModel.pb', mode='wb') as f:
f.write(output_graph_def.SerializeToString())