def _load_graph(frozen_graph_filename: "Path", prefix: str = "load", default_tf_graph: bool = False): # We load the protobuf file from the disk and parse it to retrieve the # unserialized graph_def with tf.gfile.GFile(str(frozen_graph_filename), "rb") as f: graph_def = tf.GraphDef() graph_def.ParseFromString(f.read()) if default_tf_graph: tf.import_graph_def(graph_def, input_map=None, return_elements=None, name=prefix, producer_op_list=None) graph = tf.get_default_graph() else: # Then, we can use again a convenient built-in function to import # a graph_def into the current default Graph with tf.Graph().as_default() as graph: tf.import_graph_def(graph_def, input_map=None, return_elements=None, name=prefix, producer_op_list=None) return graph
def _load_graph(self, frozen_graph_filename, prefix='load', default_tf_graph=False): # We load the protobuf file from the disk and parse it to retrieve the # unserialized graph_def with tf.gfile.GFile(frozen_graph_filename, "rb") as f: graph_def = tf.GraphDef() graph_def.ParseFromString(f.read()) if default_tf_graph: tf.import_graph_def(graph_def, input_map=None, return_elements=None, name=prefix, producer_op_list=None) graph = tf.get_default_graph() else: # Then, we can use again a convenient built-in function to import a graph_def into the # current default Graph with tf.Graph().as_default() as graph: tf.import_graph_def(graph_def, input_map=None, return_elements=None, name=prefix, producer_op_list=None) # for ii in graph.as_graph_def().node: # print(ii.name) return graph
def load_graph(graphName): graph_def = tf.GraphDef() with open(graphName, "rb") as f: graph_def.ParseFromString(f.read()) with tf.Graph().as_default() as graph: tf.import_graph_def(graph_def, name="") return graph
def convert_pb_to_pbtxt(pbfile: str, pbtxtfile: str): """Convert DP graph to graph text. Parameters ---------- pbfile : str filename of the input graph pbtxtfile : str filename of the output graph text """ with gfile.FastGFile(pbfile, 'rb') as f: graph_def = tf.GraphDef() graph_def.ParseFromString(f.read()) tf.import_graph_def(graph_def, name='') tf.train.write_graph(graph_def, './', pbtxtfile, as_text=True)
def _import_graph_def_from_frz_model(self, frz_model, feed_dict, return_elements): graph, graph_def = load_graph_def(frz_model) return tf.import_graph_def(graph_def, input_map=feed_dict, return_elements=return_elements, name="")
def load_graph(graph_name: str) -> tf.Graph: """Load graph from passed in path. Parameters ---------- graph_name : str path to frozen graph on disk Returns ------- tf.Graph tf graph object """ graph_def = tf.GraphDef() with open(graph_name, "rb") as f: graph_def.ParseFromString(f.read()) with tf.Graph().as_default() as graph: tf.import_graph_def(graph_def, name="") return graph
def load_graph_def(model_file: str) -> Tuple[tf.Graph, tf.GraphDef]: """ Load graph as well as the graph_def from the frozen model(model_file) Parameters ---------- model_file : str The input frozen model path Returns ------- tf.Graph The graph loaded from the frozen model tf.GraphDef The graph_def loaded from the frozen model """ graph_def = tf.GraphDef() with open(model_file, "rb") as f: graph_def.ParseFromString(f.read()) with tf.Graph().as_default() as graph: tf.import_graph_def(graph_def, name="") return graph, graph_def
def convert_pb_to_pbtxt(pbfile, pbtxtfile): with gfile.FastGFile(pbfile, 'rb') as f: graph_def = tf.GraphDef() graph_def.ParseFromString(f.read()) tf.import_graph_def(graph_def, name='') tf.train.write_graph(graph_def, './', pbtxtfile, as_text=True)
def _load_sub_graph(self): sub_graph_def = tf.GraphDef() with tf.Graph().as_default() as sub_graph: tf.import_graph_def(sub_graph_def, name="") return sub_graph, sub_graph_def