2017-08-09 63 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

相關問題