Easily visualize embedding on tensorboard with thumbnail images and labels.
Currently this repo is compatible with Tensorflow r1.0.1
import embedder
# create the model graph and get the last layer's output.
logits = model()
# init session and restore pre-trained model file
sess = tf.Session()
sess.run(tf.global_variables_initializer())
saver = tf.train.Saver()
saver.restore(sess, os.path.join(test_path, 'model.ckpt'))
# read your dataset and labels
data_sets, labels = read_data_sets()
# run your model
feed_dict = {input_placeholder: dataset, label_placeholder: labels}
activations = sess.run(logits, feed_dict)
embedder.summary_embedding(sess=sess, dataset=data_sets, embedding_list=[activations],
embedding_path="your embedding path", image_size=your_image_size, channel=3,
labels=labels)
import embedder
# create the model graph
logits = model()
# init session and restore pre-trained model file
sess = tf.Session()
sess.run(tf.global_variables_initializer())
saver = tf.train.Saver()
saver.restore(sess, os.path.join(test_path, 'model.ckpt'))
total_dataset = None
total_labels = None
total_activations = None
for i in range(10)
data_sets, labels = get_batch(i)
feed_dict = {input_placeholder: dataset, label_placeholder: labels}
activations = sess.run(logits, feed_dict)
if total_dataset is None:
total_dataset = data_sets
total_labels = labels
total_activations = activations
else:
total_dataset = np.append(data_sets, total_dataset, axis=0)
total_labels = np.append(labels, total_labels, axis=0)
total_activations = np.append(activations, total_activations, axis=0)
embedder.summary_embedding(sess=sess, dataset=total_dataset, embedding_list=[total_activations],
embedding_path="your embedding path", image_size=your_image_size, channel=3,
labels=total_labels)
python test_mnist.py
(python test_mnist_large_data.py)
tensorboard --log_dir=./
This should print that TensorBoard has started. Next, connect http://localhost:6006 and click the EMBEDDINGS menu.
def summary_embedding(sess, dataset, embedding_list, embedding_path, image_size, channel=3, labels=None):
pass
http://www.pinchofintelligence.com/simple-introduction-to-tensorboard-embedding-visualisation/ tensorflow/tensorflow#6322