- 微信
- 微博
  
  分享文章到微博
- 复制链接
  
  复制链接到剪贴板

CycleGAN-and-pix2pix 环境搭建：数据集下载

墨理学AI 发表于 2022/02/17 10:53:57 2022/02/17

【摘要】 CycleGAN and pix2pix in PyTorch 按照 gitHub上的教程，下载对应数据集和预训练模型

🎉 声明: 作为全网 AI 领域干货最多的博主之一，❤️ 不负光阴不负卿 ❤️

❤️【专栏：数据集整理】❤️ 之【有效拒绝假数据】

👋 Follow me 👋，一起 Get 更多有趣 AI、冲冲冲 🚀 🚀

CycleGAN and pix2pix in PyTorch


conda  create -n pix2pix python=3.6.5

pip install torch==1.1.0

pip install pillow

pip install torchvision==0.3.0

pip install dominate

pip install visdom

官方GitHub
这个项目，官方提供了非常详细的运行、测试教程，以及预训练模型，本博文就不画蛇添足了；
鼓励大家遇到项目，首先认真看一下，官方的 readMe ;
一些官方的readme不够充分的情况下，我的博文往往会起到简介、梳理、可正确运行的助力，感谢大家的支持；

测试

按照 gitHub上的教程，下载对应数据集和预训练模型之后，运行：

python test.py --dataroot datasets/horse2zebra/testA --name horse2zebra_pretrained --model test --no_dropout

发现：cycleGan-pix2pix/results 目录下生成对应的结果文件夹，查看效果即可

论文工作总结

《Image-to-Image Translation with Conditional Adversarial Networks》

《使用条件对抗网络进行图像到图像的转换》

pix2pix 简介：

这篇论文发表在CVPR2017，简称pix2pix，是将GAN应用于有监督的图像到图像翻译的经典论文，有监督表示训练数据是成对的。图像到图像翻译（image-to-image translation）是GAN很重要的一个应用方向，什么叫图像到图像翻译呢？其实就是基于一张输入图像得到想要的输出图像的过程，可以看做是图像和图像之间的一种映射（mapping），我们常见的图像修复、超分辨率其实都是图像到图像翻译的例子。
pix2pix基于GAN实现图像翻译，更准确地讲是基于cGAN（conditional GAN，也叫条件GAN），因为cGAN可以通过添加条件信息来指导图像生成，因此在图像翻译中就可以将输入图像作为条件，学习从输入图像到输出图像之间的映射，从而得到指定的输出图像。而其他基于GAN来做图像翻译的，因为GAN算法的生成器是基于一个随机噪声生成图像，难以控制输出，因此基本上都是通过其他约束条件来指导图像生成，而不是利用cGAN，这是pix2pix和其他基于GAN做图像翻译的差异。