CGAN_jittor

简介

A Jittor implementation of Conditional GAN (CGAN).

本项目是基于清华大学 Jittor（计图）框架实现的条件生成对抗网络。通过在 MNIST 数据集上进行对抗训练，模型能够根据给定的数字标签生成高质量的手写数字图像。

环境安装

本项目推荐在 Linux 环境下运行。

# 1. 克隆仓库
git clone https://www.gitlink.org.cn/lalazhuu/cgan_jittor.git
cd cgan_jittor

# 2. 安装依赖
pip install jittor==1.3.8.5
pip install matplotlib

# 3.运行方法
直接运行主程序即可开始训练并生成结果：
python CGAN.py

运行结果

损失函数曲线和最终生成结果见项目图片。