计图热身挑战赛

简介

本项目在数字图片数据集 MNIST 上训练 Conditional GAN模型，通过输入一个随机向量 z 和额外的辅助信息 y (如类别标签)，生成特定数字的图像

安装

本项目使用一张3080显卡进行训练，训练时间约为半个小时。

运行环境

Jittor 1.3.1 Python 3.8(ubuntu18.04) Cuda 11.3

安装依赖

执行以下命令安装 python 依赖

pip install -r requirements.txt

训练

运行以下命令

run CGAN.py

推理

生成的结果图片为result.png