Jittor 热身赛 CGAN

简介

本项目包含了第五届计图挑战赛计图-热身赛 Conditional GAN 的代码实现。本项目的特点是：在给定模版上正确实现了Conditional GAN的完整训练代码，成功训练出了效果很好的模型。

安装

本项目在 1 张 3080 上运行，训练时间约为 1 小时。

运行环境

• Ubuntu 22.04.3 LTS
• python >= 3.7
• jittor >= 1.3.0

安装依赖

执行以下命令安装 python 依赖

pip install -r requirements.txt

训练

单卡训练可运行以下命令：

python CGAN.py

推理

运行 CGAN.py 脚本后自动生成测试结果 result.png