文件名称:pix2pix:有条件对抗网络的图像到图像翻译的PyTorch实现
文件大小:9.42MB
文件格式:ZIP
更新时间:2024-06-07 06:25:04
Python
pix2pix 有条件对抗网络的图像到图像翻译的PyTorch实现 纸和官方代码 介绍 Pix2Pix是一个图像到图像的翻译项目,它可以做很多事情,下面仅显示其中一些: 它基于条件GAN,其中条件不是矢量或图像,而是图像。如下所示: 发电机 本文比较了两种不同的生成器,编码解码器和U-Net。 结果表明,U-Net可以做得更好,这可能是因为U-Net具有一些跳过连接,这使您可以更好地了解底层功能。 判别器 本文使用patchGAN作为判别器,这意味着我们不判断整个图像对,而是判断一些图像补丁,然后取平均值。 这样可以加快训练阶段,并可以处理不同大小的图像。 数据集 团队还会发布一些不错的数据集,您可以免费下载。 我将使用城市景观数据集。 您可以下载自己喜欢的数据集并放入数据子目录。 要求 火炬0.4.0 火炬视觉 火 我使用pytorch 0.4.0来构建此项目,因此您需要更新py
【文件预览】:
pix2pix-master
----main.py(5KB)
----images()
--------10000.png(2.09MB)
--------128500.png(1.78MB)
--------542500.png(1.85MB)
--------1.png(1.85MB)
--------2.png(71KB)
--------README.md(2B)
--------594500.png(1.77MB)
--------3.png(7KB)
----data()
--------README.md(25B)
----test.py(1KB)
----sample()
--------README.md(30B)
----checkpoints()
--------README.md(20B)
----requirements.txt(30B)
----model.py(3KB)
----config.py(691B)
----README.md(4KB)
----unet_block.py(1KB)
----data.py(1KB)