如何叠加张量图像

Question

我正在尝试做Pix2Pix NN，我的输入是灰度的，输出是RGB的，我使用

stacked_image = tf.stack([inimg, tgimg], axis=0)

为了堆叠图像以进行数据扩充，使用

inimg = tf.cast(tf.image.decode_jpeg(tf.io.read_file(INPATH + filename)), tf.float32)[..., :3]
tgimg = tf.cast(tf.image.decode_jpeg(tf.io.read_file(OUPATH + filename)), tf.float32)[..., :3]

但是我在形状上有个错误

InvalidArgumentError: Shapes of all inputs must match: values[0].shape = [286,286,1] != values[1].shape = [286,286,3] [Op:Pack] name: stack

关于如何解决这个问题的一些建议

Answer 1

如果inimg和tgimg的形状相同，则只能使用tf.stack。您可以使用tf.tile实现这一点

stacked_image = tf.stack([tf.tile(inimg, [1, 1, 3]), tgimg], axis=0)

堆叠的图像将具有形状（2、286、286、3）