Conditional GAN

计图挑战热身赛

本赛题将会提供数字图片数据集 MNIST，参赛选手需要训练一个将随机噪声和类别标签映射为数字图片的Conditional GAN模型，并生成比赛页面指定数字序列。

本赛题提供示例代码框架，提供数据下载、模型定义、训练步骤等功能。

选手可以基于示例代码填充注释为 TODO 的部分完成该赛题。

项目框架

├── CGAN.py
├── result.png

其中CGAN.py包含了模型和训练函数，result.png是模型对于随机数字字符串输入的输出结果。

环境依赖

python                 3.7
numpy                  1.24.3
jittor                 1.3.7.16
tqdm                   4.65.0

运行环境：Ubuntu Server 22.04

项目运行

python3.7 CGAN.py

生成示例