tf.summary.scalar()
用于显示标量,例如损失、精度曲线等。
tf.summary.image(name,tensor,image_number)
同于显示图片
变量含义:
name:命名
tensor:张量,通道数最大为4
image_number:输出图片数量
例如:
输出:conv的tensor为[128,28,28,32]
含义: batch_size为128,图片大小为28*28,通道数为32
用tf.summary.image()会出现错误
解决方法:
split = tf.split(conv,conv.shape[-1],axis=3) #conv.shape[-1]=32
tf.summary.image("connv",split[0],conv.shape[-1]) #split[0]为1通道