我尝试为自己的数据集修改DCGAN的示例代码enter link description here。示例代码用于RGB图像,但我的数据是灰度图像。因此,我将通道数“ nc”的值设置为1。但是,当我尝试运行该程序时,出现此错误:RuntimeError:给定groups = 1,权重为64 1 4 4,预期输入[128,3,64,64]具有1个通道,但是却有3个通道。我确实知道这是关于频道数量的问题,但是我不知道还应该更改什么来解决此错误。
任何帮助将不胜感激!
print(image.shape)
属性输入了图像形状?如果使用torchvision
(在引擎盖下带有pillow
的情况下,则可以使用torchvision.transforms.Grayscale()
(请参阅docs)。还要验证您的图像确实是灰度的。