pytorch - 彩色图像上的 GAN

Question

大多数（PyTorch）开源 GAN 在 MNIST 数据集（即灰度图像）上工作。

我可以在彩色图像的每个通道上使用 GAN，然后组合结果吗？

Answer 1

你可以让你的生成器和鉴别器生成和分类 3 通道图像——从实现的角度来说，让它们在张量上工作，B x 3 x H x W而不是在B x 1 x H x WMNIST 上工作。

你不能只在每个通道上单独使用你的 GAN 并在最后连接起来，因为你无法确保每个通道对应于相同的图像。假设您首先生成红色通道，然后生成绿色通道，最后生成蓝色通道来生成名人面孔。你如何确保红色通道没有女性样本，绿色通道没有男性样本？