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
来生成指定数字的图片。
CGAN_jittor
南开大学计算机图形学课程平时作业:计图挑战热身赛。该项目实现了一个基于 Jittor 框架的生成对抗网络(CGAN)。使用 MNIST 数据集,并通过生成和判别模型进行对抗训练。
参考环境
运行方式
使用
即可开始训练,随后,对于已经训练好的模型,可以使用
来生成指定数字的图片。