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