Jittor CGAN 手写数字生成 热身赛

Conditional GAN: 生成手写数字序列-生成注册手机号

简介

本项目包含了第二届计图挑战赛热身赛-生成手写数字图片的代码实现。

• 通过手写数字数据集训练模型，生成指定数字序列图片

安装

本项目于Macbook pro CPU训练时间约为45分钟

运行环境

• Python version: >= 3.7
• Jittor version: >= 1.3.0
  • To install Jittor, please follow the tips on the web of Jittor

安装依赖

• 为在相应系统上安装计图，请参照官网Jittor执行相应命令

训练

请通过如下命令运行本项目：

cd ./CGAN
python CGAN.py

致谢

此项目基于计图挑战赛-热身赛所提供的基本代码实现，可关注官网下载源代码。

注意事项

本项目每1000个batch输出一次中间结果图片，图片量较大，可通过修改输出防止侵占过多内存。