CGAN_jittor

南开大学计算机图形学课程平时作业：计图挑战热身赛。该项目实现了一个基于 Jittor 框架的生成对抗网络（CGAN）。使用 MNIST 数据集，并通过生成和判别模型进行对抗训练。

参考环境

• os：Ubuntu 22.04 LTS
• Python：3.10.12
• g++：11.4.0
• Jittor：1.3.9
• NVIDIA Driver：560.35.05
• CUDA：12.2.91
• cuDNN：8.9.7.29

运行方式

使用

python3 CGAN.py

即可开始训练，随后，对于已经训练好的模型，可以使用

python3 gen.py

来生成指定数字的图片。