带有keras的autoencoder的InvalidArgumentError

Question

我正在训练一个自动编码器（使用keras博客代码），我收到了错误的错误

InvalidArgumentError (see above for traceback): Incompatible shapes: [1,420,748,3] vs. [1,422,750,3]
 [[Node: loss/conv2d_7_loss/sub = Sub[T=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:CPU:0"](conv2d_7/Sigmoid, _arg_conv2d_7_target_0_1)]]

在某些搜索中，人们指出这是GPU上内存消耗过多的问题。但是我一次加载一张图片并用

进行训练

m.fit(blured, img, epochs = 1, batch_size = 1)

Answer 1

问题是你的变量“模糊”（420 x 748 x 3）和“img”（422 x 750 x 3）不具有相同的形状。自动编码器需要相同的输入和标签形状，因为它经过培训，可以学习从输入到瓶颈层，从那里到标签。