GAN 模型代码修改——3 通道到 1 通道

Question

该模型专为处理 3 通道图像 (RGB) 而设计，而我需要处理一些黑白图像数据（灰度），因此我想将“ch”参数更改为“1”而不是“3”。

完整代码可在此处获得 - https://pytorch.org/tutorials/beginner/dcgan_faces_tutorial.html

如果我们只是改变这个参数——“nc = 3” -->“nc = 1”——而不调整生成器和鉴别器的代码块，执行只会给出一个错误信息：

RuntimeError: Given groups=1, weight of size [64, 1, 4, 4], expected input[128, 3, 64, 64] to have 1 channels, but got 3 channels instead

是否有关于如何修改或使用 此公式（形状部分）手动计算这些值的指南？

请指教。

Answer 1

灰度图像是彩色图像的“特例”：如果红色通道等于绿色等于蓝色，则像素具有灰色。 因此，值为[200, 10, 30]的像素将呈绿色，而值为[180, 180, 180]的像素将呈灰色。
因此，使用预训练的 RGB 模型处理灰度图像的最简单方法是将灰度图像的单个通道复制 3 次，以生成具有灰色的三个通道的类 RGB 图像。