目录
目录readme.md

CGAN for Jittor

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

安装环境

本项目需要使用Jittor框架,具体的计图框架安装步骤请参考(https://www.educoder.net/competitions/Jittor-4)[计图框架]

运行环境

  • ubuntu >= 16.04 或 Windows Subsystem of Linux(WSL)
  • python >= 3.7
  • jittor >= 1.3.0

训练

为了训练我们的模型,你需要数字图片数据集 MNIST,直接使用一下命令来调用数据集

from jittor.dataset.mnist import MNIST

检查数据集,然后运行:

python CGAN.py

训练结束之后,程序会以‘result.png’的格式保存到文件中。 下面是生成结果图片: 图片示例

关于

Jittor-CGAN

47.0 KB
邀请码
    Gitlink(确实开源)
  • 加入我们
  • 官网邮箱:gitlink@ccf.org.cn
  • QQ群
  • QQ群
  • 公众号
  • 公众号

©Copyright 2023 CCF 开源发展委员会
Powered by Trustie& IntelliDE 京ICP备13000930号