CGAN_jittor

A Conditional GAN (CGAN) implementation based on the Jittor deep learning framework.

---

项目简介

本项目实现了一个基于 Jittor 框架的条件生成对抗网络（Conditional GAN，CGAN）。
模型可以根据输入的类别标签生成对应类别的手写数字图像（类似 MNIST 数据集风格）。
支持训练和测试，支持自定义训练参数。

---

目录结构

├── .gitignore        
├── CGAN.py          
├── result.png      
├── README.md        

---

环境依赖

前往 https://cg.cs.tsinghua.edu.cn/jittor/download/ 安装Jittor框架。

---

使用说明

python3 CGAN.py 

运行脚本后，会进行模型训练，训练完成后，会保存模型权重文件 generator_last.pkl 和 discriminator_last.pkl，并生成 result.png 图片文件，展示number对应的生成图像。

---