CGAN_jittor

程思翔 计 13 班 2021010761

项目介绍

本项目为清华大学计算机科学与技术系2023年春《计算机图形学基础》课程 PA3 仓库, 该项目使用 Jittor 机器学习框架, 在数字图片数据集 MNIST 上训练了将随机噪声和类别标签映射为数字图片的 Conditional GAN 模型, 从而生成指定数字序列对应的图片.

环境要求

python>=3.8
jittor>=1.3.7

配置好环境后, 在终端输入

python CGAN.py

即可进行模型的训练并生成指定序列对应的图片. 训练好的模型将存储在当前目录下的 *.pkl 下.

致谢

组委会提供的 Jittor 代码框架, 为编码工作带来了极大的便利.