目录
目录README.md

Jittor 计图挑战热身赛

生成结果(给定随机ID为13485041347432)  热身赛评分为0.9996

简介

本项目包含了第五届计图人工智能挑战赛——热身赛的代码实现。本项目将在数字图片数据集 ·MNIST· 上训练 Conditional GAN(Conditional generative adversarial nets)模型,通过输入一个随机向量 z 和额外的辅助信息 y (如类别标签),生成特定数字的图像。

安装

本项目可在RTX 3090显卡下运行,训练时长约1个小时。

运行环境

  • Ubuntu 20.04 LTS
  • Python >= 3.7
  • Jittor >= 1.3.1

安装jittor框架

执行以下命令安装 jittor框架

sudo apt install python3.7-dev libomp-dev
python3.7 -m pip install jittor

训练

可运行一下命令进行训练

python CGAN.py

运行上述指令即可开始训练,训练使用 Jittor 框架内置的 MNIST 数据集格式,训练过程中每一个 epoch 的模型系数将保存在 models/ 文件夹下,图片样本保存在主目录下。

推理与测试

推理包含在训练部分代码中,训练完成后进行推理,生成result.png,内容为参赛选手id的手写字体。

致谢

项目代码的Baseline由第五届计图人工智能挑战赛平台提供(https://www.educoder.net/competitions/Jittor-6

关于
52.0 KB
邀请码