Как складывать тензорные изображения

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

Вы можете использовать tf.stack, только если формы inimg и tgimg одинаковы. Вы можете достичь этого, используя tf.tile

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

stacked_image будет иметь форму (2, 286, 286, 3)