TensorFlowモデル(pb)の構造を見る

create_graph_for_tensorboard.py

import tensorflow as tf
with tf.Session() as sess:
	with tf.gfile.GFile('conv_mnist.pb', 'rb') as f:
		graph_def = tf.GraphDef()
		graph_def.ParseFromString(f.read())
		g_in = tf.import_graph_def(graph_def)
		train_writer = tf.summary.FileWriter('logs')
		train_writer.add_graph(sess.graph)


'''
python create_graph_for_tensorboard.py
tensorboard --logdir=logs
'''