本文整理汇总了Python中tensorflow.python.framework.graph_util_impl.convert_variables_to_constants方法的典型用法代码示例。如果您正苦于以下问题:Python graph_util_impl.convert_variables_to_constants方法的具体用法?Python graph_util_impl.convert_variables_to_constants怎么用?Python graph_util_impl.convert_variables_to_constants使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类tensorflow.python.framework.graph_util_impl
的用法示例。
在下文中一共展示了graph_util_impl.convert_variables_to_constants方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: freeze
# 需要导入模块: from tensorflow.python.framework import graph_util_impl [as 别名]
# 或者: from tensorflow.python.framework.graph_util_impl import convert_variables_to_constants [as 别名]
def freeze():
checkpoint_prefix = os.path.join(TMP_DIR, "saved_checkpoint")
checkpoint_state_name = "checkpoint_state"
input_graph_name = "input_graph.pb"
output_graph_name = "freezed.pb"
saver_write_version = 1
# We'll create an input graph that has a single variable containing 1.0,
# and that then multiplies it by 2.
from tensorflow.python.framework import ops
with ops.Graph().as_default():
from keras import backend as K
K.set_learning_phase(0)
model = createModel()
model.load_weights(KERAS_WEIGHTS_FILE)
sess = K.get_session()
from tensorflow.python.framework.graph_util_impl import convert_variables_to_constants
# convert_variables_to_constants(sess, sess.graph.as_graph_def(), [model.output.name.split(':')[0]])
testGraph(sess, '')
from tensorflow.python.training import saver as saver_lib
saver = saver_lib.Saver(write_version=saver_write_version)
checkpoint_path = saver.save(
sess,
checkpoint_prefix,
global_step=0,
latest_filename=checkpoint_state_name)
from tensorflow.python.framework import graph_io
graph_io.write_graph(sess.graph, TMP_DIR, input_graph_name)
sess.close()
# We save out the graph to disk, and then call the const conversion
# routine.
input_graph_path = os.path.join(TMP_DIR, input_graph_name)
input_saver_def_path = ""
input_binary = False
output_node_names = model.output.name.split(':')[0]
restore_op_name = "save/restore_all"
filename_tensor_name = "save/Const:0"
output_graph_path = os.path.join(MODEL_DATA_DIR, output_graph_name)
clear_devices = False
from tensorflow.python.tools import freeze_graph
freeze_graph.freeze_graph(input_graph_path, input_saver_def_path,
input_binary, checkpoint_path, output_node_names,
restore_op_name, filename_tensor_name,
output_graph_path, clear_devices, "")
exportWordIndex(loadWordIndex())