pytorch-CycleGAN-and-pix2pix

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

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 图像生成 🍴 复刻 6555 🐛 待处理 589 👀 关注 337
GitHub ↗
GitHub 星标 ★ 25276
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,完全免费,可自行部署使用。

项目预览

pytorch-CycleGAN-and-pix2pix 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类图像生成
开发团队junyanz
定价模式🏷️完全免费
价格说明开源项目,完全免费,可自行部署使用。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议NOASSERTION
主要语言Python
技术栈/模型computer-graphics,computer-vision,cyclegan,deep-learning,gan,gans,generative-adversarial-network,image-generation,image-manipulation,pix2pix,pytorch
GitHub 星标★ 25276
上线时间2017-04-18 00:00:00
代码更新2025-08-06
仓库体积8.3 MB
本站数据更新2026-11-09
维护状态低维护
收录时间2026-08-09
浏览次数9

项目介绍

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

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 25276

使用教程

ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。

难度:入门 约 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

上一篇:InvokeAI

下一篇:satori

同类项目推荐

UniPic 开源

一句话指令,多图联动编辑,效果媲美商业模型

Open-source SOTA multi-image editing model

★ 877 2026-08-09
mflux 开源

在 Mac 上本地跑 FLUX 等图像视频生成模型

Apple MLX native implementations of state-of-the-art generative image & video models

★ 2581 2026-11-03
onnx-web 开源

在 Windows 和 AMD 显卡上轻松跑 Stable Diffusion 的 Web 界面

web UI for GPU-accelerated ONNX pipelines like Stable Diffusion, even on Windows and···

★ 232 2026-08-09
CPPN-WGAN 开源

用CPPN+WGAN生成高质抽象艺术,探索AI创作边界

Generative Art Experiments

★ 54 2026-08-09
ComfyUI-Photopea 开源

在 ComfyUI 里直接打开 Photopea 编辑器,修完图无缝回流工作流。

Edit images in the Photopea editor directly within ComfyUI.

★ 195 2026-08-09
Conditional-GAN 开源

给GAN加个条件,想生成哪类图就生成哪类图

Tensorflow implementation for Conditional Convolutional Adversarial Networks.

★ 219 2026-08-09
semantic-draw 开源

边画边生成,实时交互式图像创作

Official code for the CVPR 2025 paper "SemanticDraw: Towards Real-Time Interactive C···

★ 587 2026-08-09
text2art 开源

输入文字,一键生成多种风格的艺术画作

AI-powered Text-to-Art Generator - Text2Art.com

★ 770 2026-08-09