create_pb.py

#create pb and pbtxt from data,index and meta

import tensorflow as tf
import tensorflow.contrib.slim as slim
print("imported Tensorflow")
tf.train.import_meta_graph('mpii-single-resnet-101.meta')
print("imported meta graph")
variables_to_restore = slim.get_variables_to_restore(include=["resnet_v1"])
restorer = tf.train.Saver(variables_to_restore)
saver = tf.train.Saver(max_to_keep=5)

tf.train.write_graph(sess.graph_def, './', 'graph.pbtxt')
tf.train.write_graph(sess.graph_def, './', 'graph.pb', as_text=False)