CGAN_Jittor

该项目是一个基于计图Jittor框架编写的，在MNIST上训练的，将随机噪声和类别标签映射为数字图片的CGAN模型，它可以生成指定数字序列对应的图片。

运行环境

Jittor的安装方式及环境要求请参考：

安装 — Jittor (tsinghua.edu.cn)

运行方式

在./CGAN.py的第199行，你可以指定一个字符串类型的数字序列，用于生成对应图片，如

number = "13707951369661"

此后，只需要在根目录下运行

python ./CGAN.py

这将会生成generator_last.pkl和discriminator_last.pkl两个模型文件，若干随机采样输入生成的图片，以及result.png。