2017-02-19 50 views
0

我試圖在Tensorboard中直接顯示由Tensorflow生成的圖像。我試圖用這個解決方案Tensorflow: How to Display Custom Images in Tensorboard (e.g. Matplotlib Plots),但我不明白如何爲概要創建Tensorflow圖形之前定義,可以將這個鏈接到培訓過程中產生的圖像:在tensorboard中顯示matplotlib圖形而不是傾銷到文件

def plot(samples): 
    fig = plt.figure(figsize=(4, 4)) 
    gs = gridspec.GridSpec(4, 4) 
    gs.update(wspace=0.05, hspace=0.05) 
    for i, sample in enumerate(samples): 
     ax = plt.subplot(gs[i]) 
     plt.axis('off') 
     ax.set_xticklabels([]) 
     ax.set_yticklabels([]) 
     ax.set_aspect('equal') 
     plt.imshow(sample.reshape(28, 28), cmap='Greys_r') 
    return fig 

    # .... 

    if it % 1000 == 0: 
     samples = sess.run(G_sample, feed_dict={z: sample_z(16, z_dim)}) 
     fig = plot(samples) 
     plt.savefig('out/{}.png' 
        .format(str(i).zfill(3)), bbox_inches='tight') 
     i += 1 
     plt.close(fig) 

回答

0

我想你混淆了兩件事情:Tensorboard摘要(https://www.tensorflow.org/api_docs/python/tf/summary/image)並保存圖像。

如果要在tensorboard中顯示樣本,則必須在代碼中使用tf.summary.image併爲其提供G_sample張量。如果你看看https://www.tensorflow.org/api_guides/python/summary#Generation_of_Summaries,你會得到更多關於如何使用文件編寫器的信息。

讓我知道這是否有幫助,或者你卡在其他地方!

+0

我的問題是,我沒有得到我怎麼能沒有通過matplotlib我的數據轉換爲圖像摘要格式。我想用一些將結果推到張量板上的plt.savefig替換。我已經能夠在張量板上顯示標量摘要。以相同的例子。 –

相關問題