ConditionalGAN

第二届计图人工智能挑战赛-计图挑战热身赛

简介

本项目依据提供的数字图片数据集 MNIST，需要训练一个将随机噪声和类别标签映射为数字图片的Conditional GAN模型，基于注释为示例代码填充 TODO 的部分完成该赛题。

安装

本项目可在显卡上运行，训练时间约为3小时。

运行环境

• window11

• python >= 3.8

• jittor >= 1.3.0

• pillow >= 6.2.1

• numpy >= 1.18.2

安装依赖

python -m pip install jittor
python -m jittor.test.test_core
python -m jittor.test.test_example
python -m jttor.test.test_cudnn_op

训练

可在虚拟环境下运行

致谢

此项目基于jittor的热身赛，部分代码参考了（https://github.com/Jittor/JGAN)

结果