La visualisation de la sortie de convolution de la couche dans tensorflow

J'essaye de visualiser la sortie d'une convolution de la couche dans tensorflow à l'aide de la fonction tf.image_summary. Je suis déjà en train de l'utiliser avec succès dans d'autres instances (e. g. la visualisation de l'image d'entrée), mais certains ont des difficultés remodelage de la sortie ici correctement. J'ai le texte suivant conv couche:

img_size = 256
x_image = tf.reshape(x, [-1,img_size, img_size,1], "sketch_image")

W_conv1 = weight_variable([5, 5, 1, 32])
b_conv1 = bias_variable([32])

h_conv1 = tf.nn.relu(conv2d(x_image, W_conv1) + b_conv1)

La sortie de h_conv1 aurait la forme [-1, img_size, img_size, 32]. Juste à l'aide de tf.image_summary("first_conv", tf.reshape(h_conv1, [-1, img_size, img_size, 1])) Ne tient pas compte de l'32 les différents noyaux, donc je suis fondamentalement de découpage à travers différentes fonctionnalités des cartes ici.

Comment puis-je les remodeler correctement? Ou est-il une autre fonction d'aide je pourrais utiliser, y compris pour cette sortie dans le résumé?

InformationsquelleAutor panmari | 2015-11-19