CGAN_jittor

Description

CGAN_jittor通过使用jittor深度学习框架，在数字图片数据集MNIST上训练一个将随机噪声和类别标签映射为数字图片的Conditional GAN模型。同时，你可以指定任意的数字序列，并用本项目生成指定数字序列对应的图片。

Installation

请先安装Jittor框架，详见https://cg.cs.tsinghua.edu.cn/jittor/download/

git clone git@code.gitlink.org.cn:LTNSXD/CGAN_jittor.git

Usage

模型训练

python CGAN.py --args [val]

可选参数详见CGAN.py

生成指定数字序列

更改CGAN.py第199行处的number为需要的数字序列。如果已经训练完毕，可以将训练部分代码注释直接生成。