| 第二届计图挑战赛开源热身赛 jittor-AutoZebra-CGAN

Jittor 计图挑战热身赛 Conditional GAN

简介

本项目包含了第二届计图挑战赛计图 开源热身赛的代码实现。本项目的特点是：使用Jittor在MNIST数据集训练CGAN模型，使用全连接模型，输入一个随机提供的向量以及类别标签，生成该数字向量的手写图像。

安装

可在CPU和GPU上运行

运行环境

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

安装依赖

执行以下命令安装 python 依赖

pip install -r requirements.txt

训练与测试

运行以下命令：

python CGAN.py

即可在同级目录下生成result.png结果文件

致谢

此项目部分代码参考了 jittor-gan。