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

简介

本项目包含了第二届计图挑战赛计图 - 热身赛比赛的代码实现。本项目采用了jittor语言构建conditional GAN网络并采用MNIST训练集进行了训练，

安装

运行环境

• Windows11
• python >= 3.8
• jittor >= 1.3.0

安装依赖

执行以下命令安装 python 依赖

pip install -m jittor

预训练模型

预训练模型在项目目录下save_model文件夹当中

训练

直接运行CGAN.py即可

推理

训练完成后运行test.py可以生成result.png

致谢

代码实现参考了jittor文档：Jittor-GAN — Jittor 1.3.4.3 文档