我正在尝试通过此代码实现WGAN-GP
https://github.com/eriklindernoren/Keras-GAN/blob/master/wgan_gp/wgan_gp.py
我想使用我自己的较大图像（384x384）和RGB而不是黑白图像的数据集，而不是默认的mnist数据集。 我尝试将通道数从1更改为3，以使颜色通道成为accoutn，但是在训练时我仍然遇到以下错误。 仍然是新事物，我不确定是否需要进行其他修改才能运行代码。
当我使用train（）
ValueError: Error when checking input: expected input_28 to have 4 dimensions, but got array with shape (32, 384, 384, 1, 3)