|
|
|
@ -1,3 +1,29 @@
|
|
|
|
|
# CGAN
|
|
|
|
|
| 第二届计图挑战赛开源模板
|
|
|
|
|
|
|
|
|
|
# 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](https://data.educoder.net/api/attachments/5149654)。
|
|
|
|
|