Отображение фигуры matplotlib в тензоре, вместо того, чтобы сбрасывать файл

Question

Я пытаюсь отобразить изображения, созданные Tensorflow непосредственно в Tensorboard. Я пытался использовать это решение Tensorflow: How to Display Custom Images in Tensorboard (e.g. Matplotlib Plots), но я не понимаю, как вы могли бы связать это с изображениями, полученных в процессе обучения, как резюме определяются перед созданием Tensorflow графа:

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) 

Answer 1

Я думаю, что вы путаете две вещи: Tensorboard резюме (https://www.tensorflow.org/api_docs/python/tf/summary/image) и сохранение изображения.

Если вы хотите отобразить свои образцы в тензометрическом шкафу, вы должны использовать tf.summary.image в своем коде и передать его вашему тензору G_sample. Если вы посмотрите на https://www.tensorflow.org/api_guides/python/summary#Generation_of_Summaries, вы получите более подробную информацию о том, как вы можете использовать Filewriters и т. Д.

Сообщите мне, если это поможет, или вы застряли в другом месте!