计图挑战热身赛 Conditional GAN

简介

本项目包含了第四届计图挑战赛计图 - 热身赛的代码实现。本项目的特点是：采用了 Conditional GAN 方法在MNIST数据集上训练模型，实现0-9手写阿拉伯数字的生成。

安装

运行环境

• ubuntu 20.04 LTS
• python >= 3.9
• jittor >= 1.3.9

训练与推理

运行一下命令进行模型训练与推理：

python CGAN.py

致谢

此项目代码在计图挑战热身赛官方提供的CGAN代码模板上修改完成。