Skip to content

Instantly share code, notes, and snippets.

@GzuPark
Forked from arafatkatze/converter.py
Created June 5, 2019 09:10
Show Gist options
  • Save GzuPark/f0386c2cbf824abc8d3e28a806cbb54f to your computer and use it in GitHub Desktop.
Save GzuPark/f0386c2cbf824abc8d3e28a806cbb54f to your computer and use it in GitHub Desktop.
# This file is useful for reading the contents of the ops generated by ruby.
# You can read any graph defination in pb/pbtxt format generated by ruby
# or by python and then convert it back and forth from human readable to binary format.
import tensorflow as tf
from google.protobuf import text_format
from tensorflow.python.platform import gfile
def pbtxt_to_graphdef(filename):
with open(filename, 'r') as f:
graph_def = tf.GraphDef()
file_content = f.read()
text_format.Merge(file_content, graph_def)
tf.import_graph_def(graph_def, name='')
tf.train.write_graph(graph_def, 'pbtxt/', 'protobuf.pb', as_text=False)
def graphdef_to_pbtxt(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.pbtxt', as_text=True)
return
graphdef_to_pbtxt('graph.pb') # here you can write the name of the file to be converted
# and then a new file will be made in pbtxt directory.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment