关于灰度图像的DCGAN(Pytorch)中通道大小的问题

问题描述 投票:0回答:2

我尝试为自己的数据集修改DCGAN的示例代码enter link description here。示例代码用于RGB图像,但我的数据是灰度图像。因此,我将通道数“ nc”的值设置为1。但是,当我尝试运行该程序时,出现此错误:RuntimeError:给定groups = 1,权重为64 1 4 4,预期输入[128,3,64,64]具有1个通道,但是却有3个通道。我确实知道这是关于频道数量的问题,但是我不知道还应该更改什么来解决此错误。

任何帮助将不胜感激!

image vision dcgan
2个回答
0
投票
验证您是否通过print(image.shape)属性输入了图像形状?

如果使用torchvision(在引擎盖下带有pillow的情况下,则可以使用torchvision.transforms.Grayscale()(请参阅docs)。还要验证您的图像确实是灰度的。


0
投票
© www.soinside.com 2019 - 2024. All rights reserved.