Jittor 计图挑战热身赛

简介

本项目包含了第三届计图挑战赛 - 计图挑战热身赛的代码实现。本项目在给出的示例代码中补充了相应代码，采用 Conditional-GAN 的方法，实现了特定数字序列的手写图片生成。

安装

本项目可在 1 张 RTX3090 上运行。

运行环境

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

安装依赖

执行以下命令安装 python 依赖

pip install -r requirements.txt

预训练模型

无

数据预处理

使用 jittor 内置的 MNIST 数据集

训练

直接运行程序 python cgan.py

推理

训练结束后，程序自动对特定数字序列生成图像，储存在 result.png 中