PA3 Conditional GAN – 手写数字生成

简介

本项目包含了第二届计图挑战赛计图热身赛题，同时也是清华大学计算机系图形学课程的第3次PA。使用CGAN生成手写数字。

安装

硬件需求: CPU或GPU均可

运行环境

• ubuntu 20.04 LTS
• python >= 3.7
• jittor >= 1.3.0

安装依赖

执行以下命令安装 python 依赖

pip install -r requirements.txt

训练及推理

运行以下命令：

bash scripts/run.sh

• 可以在scipts/run.sh中修改参数。具体参数可以参考CGAN.py中的parse_args函数。
• 可以在CGAN.py中通过更改number变量来生成不同的数字。