jittor-HaJittor-CGAN

简介

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

安装

运行环境

• Ubuntu >= 16.04 或 Windows Subsystem of Linux（WSL）
• Python：版本 >= 3.7
• jittor = 1.3.9.10

安装依赖

项目环境配置参考计图框架配置说明

运行

python CGAN.py

致谢

此项目参考了示例代码以及 jittor-gan。