save_tf_proto.py File Reference

Go to the source code of this file.

Namespaces
	save_tf_proto

Functions
def	save_tf_proto.load_model (name)

Variables
	save_tf_proto.parser = argparse.ArgumentParser(description='Save model to TF protobuf')
	save_tf_proto.help
	save_tf_proto.default
	save_tf_proto.args = parser.parse_args()
	save_tf_proto.m = load_model(args.model)
list	save_tf_proto.nnames = [n.name for n in K.get_session().graph.as_graph_def().node]
	save_tf_proto.lastnode = None
string	save_tf_proto.currentname = ''
list	save_tf_proto.output_nodes = []
	save_tf_proto.basename = n[:n.find('/')]
	save_tf_proto.minimal_graph = convert_variables_to_constants(K.get_session(), K.get_session().graph.as_graph_def(), output_nodes)
	save_tf_proto.as_text