CGAN_jittor

本项目使用jittor实现了Conditional GAN网络，并且进行了手写电话号码的生成，在头歌平台上进行了测试。

仓库里包括：

1.模型定义和训练代码

2.测试结果

3.已经训练好的模型checkpoint以及demo代码，可以直接用于生成自定义手写数字

Requirements

python 3.7+

jittor

opencv-python

numpy

Pillow

快速开始

首先安装Jittor，官方安装连接：https://cg.cs.tsinghua.edu.cn/jittor/download/

之后安装依赖环境：

pip install -r requirements.txt

训练模型

python CGAN.py

代码会自动下载数据集文件进行训练并生成手写结果。

运行测试demo

python demo.py --number 123456

此命令会加载训练好的checkpoint并生成指定的手写数字

参考论文

Conditional Generative Adversarial Nets