Jittor 挑战赛-热身赛 Conditional GAN

result.png

简介

本项目是佛山科学技术学院”CG”队在参加第三届计图人工智能挑战赛 热身赛 所使用的代码。在数字图片数据集 MNIST 上训练一个将随机噪声和类别标签映射为数字图片的Conditiona1GAN模型，并生成下方给定用户随机ID对应的数字图片结果。

安装

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

运行环境

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

安装依赖

执行以下命令安装 python 依赖 pip install -r requirements.txt

训练

运行以下命令： python CGAN.py 即可在同目录下生成 result.png 文件

致谢

感谢jittor官方提供的示例代码,基于示例代码填充注释为 TODO 的部分完成了该赛题