Jittor热身赛

项目简介

| 在数字图片数据集 MNIST 上训练 Conditional GAN（Conditional generative adversarial nets）模型，通过输入一个随机向量 z 和额外的辅助信息 y (如类别标签)，生成特定数字的图像。

安装

• python：版本 >= 3.7
• jittor >= 1.3.0 安装依赖

  pip install jittor  

  无预训练模型

生成结果

数据预处理

transform = transform.Compose([
    transform.Resize(opt.img_size),
    transform.Gray(),
    transform.ImageNormalize(mean=[0.5], std=[0.5]),
])     

推理

根据指定的数字id，生成result.png