
pytorch-CycleGAN-and-pix2pix
拍张照丢进去,自动变身油画风大片
项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 25276
使用教程
ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。
环境要求
- Python 3.11(2025 年更新后支持)
- PyTorch 2.4
- Conda 包管理器(README 提供 environment.yml)
- 可选:Docker(README 提供预构建镜像与 Dockerfile)
- 建议具备 NVIDIA GPU,多卡训练需 DDP 支持
安装与启动步骤
-
1克隆项目仓库
把官方仓库下载到本地并进入项目目录,后续所有命令都在该目录下执行。
git clone https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix cd pytorch-CycleGAN-and-pix2pix -
2创建 Conda 环境
用仓库自带的 environment.yml 一次性装好 PyTorch 及其余依赖,环境名由文件指定。
conda env create -f environment.yml -
3激活环境
激活刚创建的环境,之后运行代码都需先激活。环境名为 pytorch-img2img。
conda activate pytorch-img2img -
4Docker 方式(可选)
不想用 Conda 的用户可用官方预构建镜像和 Dockerfile,具体命令见仓库 docs/docker.md,README 未直接给出。
如何确认成功
执行 conda activate pytorch-img2img 后命令行前缀显示 pytorch-img2img,说明依赖环境已创建并激活成功。
常见问题
Q:没有装 Conda 怎么办?
A:README 只针对 Conda 用户给出 environment.yml;非 Conda 用户可自行安装 PyTorch 及其余依赖,或改用官方 Docker 镜像(见 docs/docker.md)。
Q:Docker 方式怎么运行?
A:README 说明仓库提供预构建 Docker 镜像与 Dockerfile,完整命令与用法在 docs/docker.md 页面中,需进入该文档查看。
Q:怎么用多张 GPU 训练?
A:2025 年更新后支持单机多卡 DDP,使用 torchrun --nproc_per_node=4 train.py ... 启动,具体参数需查看仓库训练说明。
Q:想复现论文里的原始结果?
A:README 指出该 PyTorch 实现效果与原始版本相当或更好,若要完全复现论文结果,请参考原始的 CycleGAN / pix2pix Torch 代码。
注意事项
- 2025 年更新后代码支持 Python 3.11 与 PyTorch 2.4,建议按此版本准备环境。
- 多卡训练命令中的省略号代表待补参数,README 节选未给出完整训练/测试命令,请查阅仓库文档。
- README 推荐了更新的 img2img-turbo(pix2pix-turbo / CycleGAN-Turbo)与 CUT 项目,速度与效果更优。
- Repl.it 用户可直接点 README 中的 Run on Repl.it 徽章在线运行。
核心亮点
- 提供CycleGAN和pix2pix两种经典图像翻译模型的完整PyTorch实现,代码结构清晰,易于学习和复现
- 支持多种训练模式(自监督、配对、非配对),并包含丰富的命令行选项,便于灵活调整实验
- 拥有详细的README、训练/测试脚本和预训练模型,社区活跃,问题响应及时
不足之处
- 文档/社区待观察
- 代码较老,部分依赖(如visdom)维护不活跃,可能在新环境下存在兼容性问题
适用场景
- 学术研究中快速实现和对比图像到图像翻译基线模型
- 风格迁移、照片增强、语义分割标签转图像等实际应用
- 教学演示GAN架构和循环一致性损失原理
替代项目
CycleGAN-Torch、pix2pixHD、SPADE
同类项目推荐
UniPic
开源
一句话指令,多图联动编辑,效果媲美商业模型
Open-source SOTA multi-image editing model
mflux
开源
在 Mac 上本地跑 FLUX 等图像视频生成模型
Apple MLX native implementations of state-of-the-art generative image & video models
onnx-web
开源
在 Windows 和 AMD 显卡上轻松跑 Stable Diffusion 的 Web 界面
web UI for GPU-accelerated ONNX pipelines like Stable Diffusion, even on Windows and···
CPPN-WGAN
开源
用CPPN+WGAN生成高质抽象艺术,探索AI创作边界
Generative Art Experiments
On-Device-Stable-Diffusion
开源
在手机上跑 Stable Diffusion,用 Flutter 实现本地 AI 绘画
On Device Stable Diffusion In Mobile Devices
ComfyUI-Photopea
开源
在 ComfyUI 里直接打开 Photopea 编辑器,修完图无缝回流工作流。
Edit images in the Photopea editor directly within ComfyUI.
Conditional-GAN
开源
给GAN加个条件,想生成哪类图就生成哪类图
Tensorflow implementation for Conditional Convolutional Adversarial Networks.
semantic-draw
开源
边画边生成,实时交互式图像创作
Official code for the CVPR 2025 paper "SemanticDraw: Towards Real-Time Interactive C···
text2art
开源
输入文字,一键生成多种风格的艺术画作
AI-powered Text-to-Art Generator - Text2Art.com