目录
CentaureaHO3个月前4次提交
目录README.md

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

来生成指定数字的图片。

关于

A Jittor implementation of Conditional GAN (CGAN).

47.0 KB
邀请码