2017-08-09 41 views
1

我只看到其目前居住在象征性的张量(logits,标签)的图像:如何在张量板中显示我的所有图像?

with tf.name_scope("Train"): 
    optimizer = tf.train.AdamOptimizer(FLAGS.learning_rate).minimize(cost_function) 
    tf.summary.image('logits', tn_logits, max_outputs=4) 
    tf.summary.image('label', t_label, max_outputs=4) 

在会话中,我在一个循环中进给网络图像。

for epoch in range(FLAGS.training_epochs): 

     for img in images: 
      _, summary_str, costs = sess.run([optimizer, merged_summary_op, cost_function], 
              feed_dict={t_im0: img.l_img.eval(), t_im1: img.r_img.eval(), 
                 t_label: img.mask.eval()}) 

如何同时显示所有图像?


我想有这种说法对我的所有图像就像在画廊: example

+0

如果'tn_logits'和't_label'是形状为[[batch_size,height,width,channels]'的张量,张量板中将会有4个('max_outputs = 4')图像。你有不同的结果还是想要别的东西? – eaksan

+0

是的,它们的形状是[1,h,w,1]。我刚刚尝试了4个结果,但它始终保持为2. – j35t3r

回答

2

图像张量和tf.summary.imagemax_output参数的第一个维度tensorboard库中定义的图像数量。由于您一次写入1张图像,现有图像将被覆盖。

而不是迭代,连接4个图像,使得tn_logitst_label将具有[4, h, w, 1]的形状。

然后在tensorboard你将有Train/logits/image/0Train/label/image/1Train/label/image/2Train/label/image/3条目tn_logits

+0

如何连接? – j35t3r

+0

你可以使用'numpy.concatenate(images,axis = 0)':https://docs.scipy.org/doc/numpy/reference/generated/numpy.concatenate.html#numpy.concatenate – eaksan

+0

如果批次大小不同:ConcatOp:输入的尺寸应该匹配:形状[0] = [1,246,381,3]与形状[1] = [1,252,367,3] – j35t3r

相关问题