jittor 手写数字生成 CGAN

简介

本项目包含了第五届计图挑战赛计图 - 手写数字生成比赛的代码实现。本项目的特点是：采用了CGAN方法对MNIST数据集处理，取得了生成给定数字序列对应的图片的效果。

安装

本项目可在1张4050上运行，训练时间约为20min

运行环境

• python 3.10.2
• jittor 1.3.9.14

安装依赖

执行以下命令安装python依赖

pip install -r requirements.txt

预处理模型

预训练模型为MNIST数据集，无需下载

训练

可运行以下命令

python3 CGAN.py