CGAN_jittor

项目简介

使用Jittor 机器学习框架，在数字图片数据集MNIST上训练一个将随机噪声和类别标签映射为数字图片的Conditional GAN 模型，生成指定数字序列对应的图片。 更多Jittor资讯，可以参考https://cg.cs.tsinghua.edu.cn/jittor/。

环境要求

1. Windows或Linux
2. Python >= 3.7
3. g++或clang

运行方式

1. 使用pip工具链安装Jittor库

   sudo apt install python3-dev libomp-dev
   python3 -m pip install jittor

2. 在CGAN.py第200行输入你希望生成的数字序列字符串

   number = '' #写入数字序列（字符串类型）

3. 使用python运行CGAN.py

   python3 CGAN.py

   加上--n_epochs参数可修改训练次数。更多参数见CGAN.py第13-23行。