[PYTHON] I tried to visualize Google's general object recognition NN, Inception-v3 with Tensorboard
How to view the graph of Google's general object recognition network Inception on Tensorboard
[Learning library has been released] of Google's general object recognition NN, Inception-v3 released in 2015 (http://googleresearch.blogspot.jp/2016/03/train-your-own-image-classifier- with.html), so I tried to visualize the network with Tensorboard to deepen my understanding.
Prerequisites
- Python (version 2.7.10)
- Tensorflow (version 0.8.0)
procedure
- Preparation (creation of working directory)
$ mkdir /tmp/imagenet
$ cd /tmp/imagenet
- Download the trained Inception network
$ wget http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz
$ tar xzvf inception-2015-12-05.tgz
- Dump graph structure from trained Inception network
- Installation of dump script
$ vi dump.py
import os
import os.path
import tensorflow as tf
from tensorflow.python.platform import gfile
INCEPTION_LOG_DIR = '/tmp/inception_v3_log'
if not os.path.exists(INCEPTION_LOG_DIR):
os.makedirs(INCEPTION_LOG_DIR)
with tf.Session() as sess:
model_filename = '/tmp/imagenet/classify_image_graph_def.pb'
with gfile.FastGFile(model_filename, 'rb') as f:
graph_def = tf.GraphDef()
graph_def.ParseFromString(f.read())
_ = tf.import_graph_def(graph_def, name='')
writer = tf.train.SummaryWriter(INCEPTION_LOG_DIR, graph_def)
writer.close()
- Dump graph
$ mkdir /tmp/inception_v3_log
$ python dump.py
- Check the graph on Tensorboard
- Start Tensorboard
$ tensorboard --logdir /tmp/inception_v3_log/
- Access
http: // localhost: 6006 /
Reference material
- https://www.tensorflow.org/versions/r0.8/tutorials/image_recognition/index.html
- https://github.com/tensorflow/tensorflow/issues/1287
Display example
Recommended Posts