CycleGAN

环境要求

• Torch7（Lua 深度学习框架），需按官方 distro 仓库说明安装
• luarocks 可用（随 Torch 安装），用于安装 nngraph、class、display 三个包
• 数据目录下需含 trainA 与 trainB 两个子目录，分别存放 A、B 两个域的图像
• 可选：testA / testB 子目录存放测试图像
• 两个域的图像视觉内容应相似（如马↔斑马），否则效果很差

安装与启动步骤

1. 1安装 Torch7

   按官方 distro 仓库 https://github.com/torch/distro 的说明安装 Torch 及其依赖，这是本项目的核心运行环境

2. 2安装 Lua 依赖包

   用 luarocks 依次安装 nngraph、class 与 display；display 用于可视化训练过程，缺一不可

   luarocks install nngraph
   luarocks install class
   luarocks install https://raw.githubusercontent.com/szym/display/master/display-scm-0.rockspec

   复制

3. 3克隆项目仓库

   把 CycleGAN 源码下载到本地并进入项目目录，后续脚本都在该目录下运行

   git clone https://github.com/junyanz/CycleGAN
   cd CycleGAN

   复制

4. 4准备训练数据

   在 data 下建数据集文件夹，再建 trainA 和 trainB 两个子目录，分别放入两个域的图片；有测试图可另建 testA、testB

关键配置

如何确认成功

执行 th 能正常进入 Torch 交互环境，且 data 目录下 trainA、trainB 中均已放入图像，即环境与数据准备就绪。

常见问题

注意事项

• Torch7 已停止维护，安装依赖较繁琐，新手建议改用官方 PyTorch 版本。
• README 顶部推荐了后续工作 CUT（contrastive-unpaired-translation），训练更快、更省显存。
• 数据集组合不要随意搭配，优先选择内容相近的两个域。