
Colorization_GAN: 基于条件生成对抗网络的灰度图像上色方法
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本文介绍了一种基于条件生成对抗网络(Colorization_GAN)的方法,专门用于将灰度图像转换为彩色图像,以提高图像的视觉效果和信息量。
着色_GAN 使用条件生成对抗网络来对灰度图像进行上色处理。这是DCGAN的PyTorch实现,如相关论文所述。在传统的GAN中,发生器的输入是随机产生的噪声数据z。然而,由于其性质的原因,这种方法不适用于自动着色问题。因此需要修改发生器以接受灰度图像作为输入而非噪声。
通过使用一种称为条件生成对抗网络(cGAN)的方法解决了上述问题:该方法没有引入额外的噪声,而是将生成器的输入视为零噪声,并且把灰阶图像当作先验信息。鉴别器则从生成器和原始数据中获取彩色图片,并以灰度图作为参考来判断哪一张是真正的彩色照片。
网络架构方面,发生器的设计受到了U-Net结构的影响:模型具有对称设计,包括n个编码单元以及同样数量的解码单元。为了区分起见,我们采用类似的体系结构作为基线收缩路径。
数据集部分使用了CIFAR-10 数据库来进行训练和测试。 若要进行全数据集模式培养,请先下载该数据库。
全部评论 (0)
还没有任何评论哟~


