tf.summary.imageのやり方を忘れがちなのでメモしておく

image.py

import tensorflow as tf

def cnn(image):
  with tf.name_scope('conv1'):
    W_conv1 = tf.nn.conv2d(image, weight, strides=[1, 1, 1, 1], padding='SAME')
    w_conv1 = tf.transpose(tf.reshape(W_conv1, [-1, 224, 224, 3]), perm=[3, 0, 1, 2])  # 可視化用
  
    with tf.device('/cpu:0'):  # どこかでtf.device('/gpu:2')などしてると落ちるので
      tf.summary.image('filter', w_conv1, 10)
      
  return y