Jittor 计图挑战热身赛 CGAN

简介

本项目包含了第二届计图挑战赛计图 - 热身赛的代码实现。本项目的特点是训练一个将随机噪声z和类别标签映y射为数字图片的Conditional GAN模型，在数据集 MNIST 上训练，并生成注册时绑定的手机号。

安装

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

运行环境

• Windows 10 专业版（64位）
• python >= 3.7
• jittor >= 1.3.0

安装依赖

执行以下命令安装 python 依赖

pip install -r requirements.txt

模型

生成器训练模型模型在 generator_last.pkl ；

鉴别器训练模型在 discriminator_last.pkl。

训练

运行 CGAN.py 开始训练，并等待程序运行结束输出结果。

致谢

此项目基于计图挑战赛baseline实现，代码参考了 示例代码。