pytorch-CycleGAN-and-pix2pix

拍张照丢进去,自动变身油画风大片

PyTorch 实现的经典图像到图像转换框架,包含 pix2pix 与 CycleGAN 两大模型,支持风格迁移、图像修复、超分辨率等任务,是学习与复现 GAN 图像翻译的权威开源项目。

开源 free 图像生成
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 25250
维护状态 低维护
是否开源 是
定价模式 free

项目数据

分类图像生成
开发团队junyanz
所属国家
官网地址
定价模式free
价格说明开源项目,完全免费,可自行部署使用。
访问状态
是否开源是
开源协议NOASSERTION
主要语言Python
技术栈/模型computer-graphics,computer-vision,cyclegan,deep-learning,gan,gans,generative-adversarial-network,image-generation,image-manipulation,pix2pix,pytorch
GitHub 星标★ 25250
30天Star增速
HF 下载量
上线时间2017-04-18 00:00:00
最近更新2026-09-22 00:00:00
维护状态低维护
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数7

使用教程

难度:入门 约 15 分钟 部署方式:本地安装 4 步

环境要求

  • Python 3.11(2025 年更新后支持)
  • PyTorch 2.4
  • Conda 包管理器(README 提供 environment.yml)
  • 可选:Docker(README 提供预构建镜像与 Dockerfile)
  • 建议具备 NVIDIA GPU,多卡训练需 DDP 支持

安装与启动步骤

  1. 1克隆项目仓库

    把官方仓库下载到本地并进入项目目录,后续所有命令都在该目录下执行。

    git clone https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix
    cd pytorch-CycleGAN-and-pix2pix
  2. 2创建 Conda 环境

    用仓库自带的 environment.yml 一次性装好 PyTorch 及其余依赖,环境名由文件指定。

    conda env create -f environment.yml
  3. 3激活环境

    激活刚创建的环境,之后运行代码都需先激活。环境名为 pytorch-img2img。

    conda activate pytorch-img2img
  4. 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

项目介绍

pytorch-CycleGAN-and-pix2pix 是图像领域的开源项目,由 junyanz 开发,2017 年首次发布。

在全站 13,090 个收录项目中,它的 GitHub 星标数(25,250)位列前 2%,在图像分类的 839 个项目里位列前 1%。

近 41 天,它的 GitHub 星标从 25,218 增加到 25,250,净增 32。

项目已超过三个月没有代码更新,维护节奏明显放缓,最近一次代码更新于 2026-09-22。完全免费,可自行部署使用。

它主要面向的使用场景是:学术研究中快速实现和对比图像到图像翻译基线模型。同类可对比的替代方案包括 CycleGAN-Torch、pix2pixHD、SPADE。

上一篇:InvokeAI

下一篇:satori

同类项目推荐

satori 开源

网页代码一键变高清图,放大不糊,不用截图

Enlightened library to convert HTML and CSS to SVG

★ 13967 2026-08-09
C2GAN 开源

用关键点精准控制图像生成,让AI画出你想要的结构

[ACM MM 2019 Oral] Cycle In Cycle Generative Adversarial Networks for Keypoint-Guide···

★ 70 2026-08-09
ComfyUI 开源

像搭积木一样连节点,生图流程自己搭

The most powerful and modular diffusion model GUI, api and backend with a graph/node···

★ 134493 2026-08-09