计图挑战热身赛 CGAN

简介

本项目包含了第三届计图人工智能挑战赛-计图挑战热身赛的代码实现。本项目的特点是：采用了示例代码的CGAN模型对MNIST数据集进行训练，实现了生成给定ID的数字图片结果。

安装

本项目使用一张NVIDIA GeForce RTX 2070显卡进行训练

运行环境

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

安装依赖

执行以下命令安装 python 依赖

pip install -r requirements.txt

训练

运行以下命令：

python main.py

推理

训练生成的结果图为result.png