InvalidArgumentError: Tensor must be 4-D with last dim 1, 3, or 4, not [shape]问题:
在我的程序里有这个的一个函数:
tf.summary.image()
(一)具体讲一下tf.summary.image()函数
tf.summary.image(
name,
tensor,
max_outputs=3,
collections=None,
family=None
)
定义于:tensorflow/python/summary/summary.py
输出Summary带有图像的协议缓冲区。
构建的图像的Tensor必须是4-D形状[batch_size, height, width, channels], 其中channels可以是:
1: tensor被解析成灰度图像。 #1通道
3:tensor被解析成RGB图像。 #3通道
4:tensor被解析成RGBA图像。 #4通道
图像具有与输入张量相同的通道数。对于浮点输入,将值一次标准化为一个在[0, 255], uint8值不变。op使用两种不同的规范化算法:
如果输入值都是正数,则重新调整它们,因此最大值为255。
如果任何输入值为负,则移动值,使输入值0.0为127.然后重新调整