Jittor 挑战热身赛 CGAN

简介

本项目包含了第五届计图挑战热身赛计图 - CGAN的代码实现。本项目的特点是：给定数字图片数据集 MNIST，训练一个将随机噪声和类别标签映射为数字图片的Conditional GAN模型，并生成指定数字序列。

安装

本项目可在 1 张 4060 上运行，训练时间约为 5 分钟。

运行环境

• ubuntu 22.04 LTS
• python >= 3.7
• jittor >= 1.3.0

安装依赖

参考计图框架

运行方法

python CGAN.py

结果展示

训练中生成的图像和最终生成的指定数字序列图像将保存在项目根目录下。

致谢

此项目基于第五届计图挑战热身赛计图 - CGAN的赛题说明-示例代码实现。