我尝试为我自己的数据集修改 DCGAN 的示例代码enter link description here。示例代码适用于 RGB 图像，但我的数据是灰度图像。因此，我将通道数“nc”的值设置为 1。但是，当我尝试运行程序时，我得到了这个错误： RuntimeError: 给定组=1，大小为 64 1 4 4 的权重，预期输入 [128, 3, 64, 64] 有 1 个通道，但有 3 个通道。我确实知道这是关于频道数量的问题，但我不知道我应该改变什么来解决这个错误。
任何帮助将不胜感激！
【问题讨论】：
您正在传递带有3 通道的RGB 图像。你应该传递形状为 [batch, 1, width, height] 的张量，而你的显然是 [128, 3, 64, 64]（并且应该是 [128, 1, 64, 64]）。
通过print(image.shape)属性验证您输入的图像形状`
如果您使用torchvision（在后台使用pillow），则可以使用torchvision.transforms.Grayscale()（请参阅docs）。还要验证您的图像确实是灰度的。
【讨论】：
您输入的实际形状是什么？我的猜测是您将图像加载为标准 RGB，即使它们是真正的灰度图像，也会将它们加载为 3 通道图像。
【讨论】：