| 第二届计图挑战赛开源模板

Jittor 热身赛利用CGAN生成MNIST手写数字

简介

本项目包含了第二届计图热身赛计图 - CGAN生成手写数字实现的代码实现。本项目的特点是：采用了 CGAN对MNIST数据集进行处理，生成队伍号码并达到一定相似度的效果。

安装

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

运行环境

• ubuntu18.04
• python 3.8
• jittor 1.3.1

安装依赖

执行以下命令安装 python 依赖

pip install -r requirements.txt

训练

直接运行CGAN.py即可

致谢

此项目基于论文 Conditional Generative Adversarial Nets 实现，部分代码参考了 jittor-CGAN。