Jittor 数字生成热身赛

简介

本项目包含了第三届计图挑战赛热身赛计图 - MNIST的代码实现。

安装

运行环境

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

安装依赖

可参考JGAN https://github.com/Jittor/JGAN

如果数据集下载不成功可以手动下载，在报错里有网址，下载到同一目录，共三个文件。 单卡训练可运行以下命令：

python acgan.py

推理

可以修改文件中的num_list改成想要生成的数字。 生成手写数字可以运行以下命令：

python gen.py

致谢

此项目基于ACGAN论文 Auxiliary Classifier Generative Adversarial Network 实现，部分代码参考了 JGAN。