Error when parsing graph_def from string

前端 未结 2 926
谎友^
谎友^ 2020-12-30 12:39

I am trying to run a very simple saving of a Tensorflow graph as .pb file, but I have this error when parsing it back:

Traceback (most recent call last):
  F         


        
2条回答
  •  被撕碎了的回忆
    2020-12-30 13:01

    graph_def = tf.GraphDef()
    graph_def.ParseFromString(fileContent)
    

    Here fileContent should be a **"Frozen Graph". Tensorflow provides an api for the same as well, refer Tensorflow freeze_graph API

    Another way to create frozen graph is:

    with tf.Session(graph=tf.Graph()) as sess:
          saver = tf.train.import_meta_graph(<.meta file>)
          saver.restore(sess, )
          output_graph_def = tf.graph_util.convert_variables_to_constants(
                        sess,
                        tf.get_default_graph().as_graph_def(),
                        [comma separated output nodes name]
                    ) 
          # Saving "output_graph_def " in a file and generate frozen graph.
          with tf.gfile.GFile('frozen_graph.pb', "wb") as f:
          f.write(output_graph_def.SerializeToString())
    

    Use frozen_graph.pb as

    graph_def.ParseFromString("frozen_graph.pb")
    

    So first using Saver object generates the .meta and other files. Once its done create frozen graph.

提交回复
热议问题