我正在尝试做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)),:3]
但是我发现形状有误。
InvalidArgumentError: Shapes of all inputs must match: values[0].shape = [286,286,1] != values[1].shape = [286,3] [Op:Pack] name: stack
有关如何解决此问题的一些建议?