From b1f1fb0e1f2d3ccbeee661d04fe0b7f53041c3d6 Mon Sep 17 00:00:00 2001
From: Hao <qfgaohao@gmail.com>
Date: Mon, 18 Jun 2018 13:55:55 +1200
Subject: [PATCH] add code to visualize tensorfolow models in tensorboard.

---
 visual_tf_models.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 visual_tf_models.py

diff --git a/visual_tf_models.py b/visual_tf_models.py
new file mode 100644
index 00000000..f0c906fc
--- /dev/null
+++ b/visual_tf_models.py
@@ -0,0 +1,21 @@
+import tensorflow as tf
+from tensorflow.python.platform import gfile
+import sys
+import time
+
+if len(sys.argv) < 2:
+    print("Usage: python visual_tf_model.py <model.pb>")
+    sys.exit(0)
+
+model_file_name = sys.argv[1]
+with tf.Session() as sess:
+    with gfile.FastGFile(model_file_name, 'rb') as f:
+        graph_def = tf.GraphDef()
+        graph_def.ParseFromString(f.read())
+        g_in = tf.import_graph_def(graph_def)
+LOGDIR='log'
+train_writer = tf.summary.FileWriter(LOGDIR)
+train_writer.add_graph(sess.graph)
+
+while True:
+    time.sleep(1000)
\ No newline at end of file