目录
目录README.md

CGAN_jittor

本项目的功能为通过使用 Jittor 机器学习框架,在数字图片数据集 MNIST 上训练一个将随机噪声和类别标签映射为数字图片的 Conditional GAN 模型,生成指定数字序列对应的图片。

环境配置与依赖

本项目使用了 Jittor 机器学习框架,因此使用前需完成 Jittor 的安装,下载地址为https://cg.cs.tsinghua.edu.cn/jittor/download/ 其他依赖可通过 pip install -r requirements.txt 完成安装

项目运行

运行本项目前,需要对CGAN.py中的代码进行修改,将代码203行 number = 后面的序列修改为想要生成的数字序列的字符串。之后运行python CGAN.py命令即可运行本项目。

关于

A Jittor implementation of Conditional GAN (CGAN)

29.0 KB
邀请码