pix2pix
有条件对抗网络的图像到图像翻译的PyTorch实现
纸和官方代码
介绍
Pix2Pix是一个图像到图像的翻译项目,它可以做很多事情,下面仅显示其中一些:
它基于条件GAN,其中条件不是矢量或图像,而是图像。如下所示:
发电机
本文比较了两种不同的生成器,编码解码器和U-Net。 结果表明,U-Net可以做得更好,这可能是因为U-Net具有一些跳过连接,这使您可以更好地了解底层功能。
判别器
本文使用patchGAN作为判别器,这意味着我们不判断整个图像对,而是判断一些图像补丁,然后取平均值。 这样可以加快训练阶段,并可以处理不同大小的图像。
数据集
团队还会发布一些不错的数据集,您可以免费下载。
我将使用城市景观数据集。 您可以下载自己喜欢的数据集并放入数据子目录。
要求
火炬0.4.0
火炬视觉
火
我使用pytorch 0.4.0来构建此项目,因此您需要更新py
2024-04-15 23:08:01
9.42MB
Python
1