使用Jittor实现Conditional GAN

简介

本项目基于python使用jittor深度学习框架实现了条件生成对抗网络（Conditional Generative Adversarial Network, CGAN）。CGAN是一种生成对抗网络（GAN）的扩展，它在生成图像的同时，通过引入条件信息，使得生成的图像可以满足特定的条件。本项目是在数字图片数据集MNIST上训练了一个将随机噪声和类别标签映射为图片的Conditional GAN模型，可以生成指定数字序列对应的图片。

安装

安装jittor

可以执行一下指令安装jittor

sudo apt install python3.7-dev libomp-dev
python3.7 -m pip install jittor
python3.7 -m jittor.test.test_example

[安装教程] https://cg.cs.tsinghua.edu.cn/jittor/

训练模型

参考指令：

python CGAN.py