Protobuf grafiği ikili tel biçimine nasıl dönüştürülür?

Question

insan okunabilir formata ikili tel biçimine dönüştürmek için bir yöntem var ama ben sadece bunun için dosya adını yazmak zorunda bu

import tensorflow as tf 
from tensorflow.python.platform import gfile 

def converter(filename): 
    with gfile.FastGFile(filename,'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, 'pbtxt/', 'protobuf.pb', as_text=True) 
    return 

tersini yapamaz ve işe yarıyor. Ama tam tersini yaparak i

File "pb_to_pbtxt.py", line 16, in <module> 
    converter('protobuf.pb') # here you can write the name of the file to be converted 
    File "pb_to_pbtxt.py", line 11, in converter 
    graph_def.ParseFromString(f.read()) 
    File "/usr/local/lib/python2.7/dist-packages/google/protobuf/message.py", line 185, in ParseFromString 
    self.MergeFromString(serialized) 
    File "/usr/local/lib/python2.7/dist-packages/google/protobuf/internal/python_message.py", line 1008, in MergeFromString 
    if self._InternalParse(serialized, 0, length) != length: 
    File "/usr/local/lib/python2.7/dist-packages/google/protobuf/internal/python_message.py", line 1034, in InternalParse 
    new_pos = local_SkipField(buffer, new_pos, end, tag_bytes) 
    File "/usr/local/lib/python2.7/dist-packages/google/protobuf/internal/decoder.py", line 868, in SkipField 
    return WIRETYPE_TO_SKIPPER[wire_type](buffer, pos, end) 
    File "/usr/local/lib/python2.7/dist-packages/google/protobuf/internal/decoder.py", line 838, in _RaiseInvalidWireType 
    raise _DecodeError('Tag had invalid wire type.') 

Answer 1

Sen google.protobuf.text_format modülü kullanarak ters çeviri yapabilirsiniz olsun: Sen tf.Graph.as_graph_def() kullanabilirsiniz

import tensorflow as tf 
from google.protobuf import text_format 

def convert_pbtxt_to_graphdef(filename): 
    """Returns a `tf.GraphDef` proto representing the data in the given pbtxt file. 

    Args: 
    filename: The name of a file containing a GraphDef pbtxt (text-formatted 
     `tf.GraphDef` protocol buffer data). 

    Returns: 
    A `tf.GraphDef` protocol buffer. 
    """ 
    with tf.gfile.FastGFile(filename, 'r') as f: 
    graph_def = tf.GraphDef() 

    file_content = f.read() 

    # Merges the human-readable string in `file_content` into `graph_def`. 
    text_format.Merge(file_content, graph_def) 
    return graph_def 

Answer 2

ardından Protobuf en SerializeToString() şöyle:

proto_graph = # obtained by calling tf.Graph.as_graph_def() 

with open("my_graph.bin", "wb") as f: 
    f.write(proto_graph.SerializeToString()) 

---

Sadece dosyayı yazmak ve umurunda değilsen kodlaması hakkında da tf.train.write_graph()

v = tf.Variable(0, name='my_variable') 
sess = tf.Session() 
tf.train.write_graph(sess.graph_def, '/tmp/my-model', 'train.pbtxt') 

Not kullanabilirsiniz: önceki sürümleri hakkında emin TF 0.10, üzerinde test edilmiştir.