tf.summary.image执行官方示例触发Invalid Argument Error问题咨询

你的代码报错的核心原因是**tf.summary.image对输入格式的要求理解有误**：

• 当你传入列表[image1, image2]时，TensorFlow会将其转换为一个4D张量[2,8,8,1]，但tf.summary.image要求列表中的每个元素是独立的4D批次张量（形状为[batch_size, 高度, 宽度, 通道数]），而非3D的单张图像。
• 内部处理时，这个4D张量被错误地当作单张图像传给EncodePng操作，而EncodePng仅接受3D张量（[高度, 宽度, 通道数]），因此触发形状不匹配的报错。

修正后的代码 #

import tensorflow as tf

w = tf.summary.create_file_writer('test/logs')
with w.as_default():
  image1 = tf.random.uniform(shape=[8, 8, 1])
  image2 = tf.random.uniform(shape=[8, 8, 1])
  # 将两张3D图像堆叠为4D批次张量：[2, 8, 8, 1]
  images_batch = tf.stack([image1, image2], axis=0)
  tf.summary.image("grayscale_noise", images_batch, step=0)

额外说明 #

如果需要传入多个独立的图像序列（而非同一批次），可以给每张3D图像单独增加批次维度，再组成列表传入：

tf.summary.image("grayscale_noise", [tf.expand_dims(image1, 0), tf.expand_dims(image2, 0)], step=0)

内容的提问来源于stack exchange，提问作者Manish