Jittor CGAN

使用Conditional GAN算法生成手写数字
简介
本项目包含了第五届计图挑战赛(热身赛)的代码实现。本项目的特点是:采用了Conditional GAN算法,在MNIST数据集上训练,实现了手写数字的生成。
安装
本项目需要在NVIDIA GPU上运行,且需要支持cuda。
运行环境
- python >= 3.7
- jittor >= 1.3.0
安装依赖
参照cg.cs.tsinghua.edu.cn/jittor/download/
训练
运行命令:python CGAN.py [arguments]
Jittor CGAN
使用Conditional GAN算法生成手写数字
简介
本项目包含了第五届计图挑战赛(热身赛)的代码实现。本项目的特点是:采用了Conditional GAN算法,在MNIST数据集上训练,实现了手写数字的生成。
安装
本项目需要在NVIDIA GPU上运行,且需要支持cuda。
运行环境
安装依赖
参照cg.cs.tsinghua.edu.cn/jittor/download/
训练
运行命令:
python CGAN.py [arguments]