Jittor CGAN数字生成热身赛

简介

本项目包含了第五届计图挑战赛热身赛比赛的代码实现。本项目的特点是：采用了在MNIST数据集上利用CGAN进行手写数字生成。

安装

本项目可在 1 张 2080 ti 上运行，训练时间约为 20分钟。

运行环境

• ubuntu 20.04 LTS
• python >= 3.7
• jittor >= 1.3.0

安装依赖

执行以下命令安装 python 依赖

sudo apt install python3.7-dev libomp-dev
python3.7 -m pip install jittor
# or install from github(latest version)
# python3.7 -m pip install git+https://github.com/Jittor/jittor.git
python3.7 -m jittor.test.test_example
pip install numpy

训练

｜ 介绍模型训练的方法

训练可运行以下命令：

python CGAN.py

推理

生成测试集上的结果可以运行以下命令：

python CGAN.py

致谢

此项目参考第五届计图人工智能挑战赛所给的baseline进行实现

注意事项

点击项目的“设置”，在Description一栏中添加项目描述，需要包含“jittor”字样。同时在Topics中需要添加jittor。