彩色图像上的GAN

时间:2019-02-01 17:59:39

标签: pytorch generative-adversarial-network

大多数(PyTorch)open source GANs work on MNIST dataset,即灰度图像。

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

1 个答案:

答案 0 :(得分:2)

您只需让生成器和鉴别器生成并分类3通道图像-就实现而言,使它们在B x 3 x H x W张量而不是B x 1 x H x W上工作,就像对MNIST一样。< / p>

您不能仅在每个通道上单独使用GAN并在末尾进行连接,因为您将无法确保每个通道都对应于同一图像。假设您通过首先生成红色通道,然后生成绿色,最后生成蓝色来生成名人面孔。如何确定红色通道没有雌性样品,绿色通道没有雄性样品?