CGAN

简介

• 本项目包含了第三届计图挑战赛——计图挑战热身赛的代码实现。本项目基于官方提供样例代码

安装

• 本项目可在 1 张 TITAN RTX 上运行，训练时间不到 1 小时

运行环境

• Ubuntu 18.04.5 LTS
• Python >= 3.7
• jittor >= 1.3.0

安装依赖

• git clone https://github.com/Jittor/jittor.git
  pip install ./jittor
  export cc_path="g++"
  python -m jittor_utils.install_cuda
  python -m jittor.test.test_example

预训练模型

• 无

训练&推理

• python CGAN.py

• 运行结束时可看到生成器和判别器的模型文件、验证图片文件和结果图片文件

• 生成器和判别器的模型文件可通过云盘下载，云盘链接：https://cloud.tsinghua.edu.cn/d/78d656ab0946424e9643/