Jittor计图挑战热身赛-CGAN生成电话号码

简介

本项目包含了第二届计图挑战赛-热身赛的代码实现。本项目的特点是：采用CGAN算法，用MNIST数据集进行训练，得到手写数字生成器，生成指定的电话号码。

安装

本项目可在 1 张 TITAN Xp 上运行，下载数据集、编译和训练共需要约半小时。

运行环境

• ubuntu 20.04 LTS
• python >= 3.7
• jittor >= 1.3.0

安装依赖

执行以下命令安装 python 依赖

pip install jittor numpy

训练并得到最终结果

可运行以下命令，训练并得到最终生成的电话号码13245678900的手写图：

python CGAN.py --phone 13245678900

致谢

此项目基于官方提供的baseline实现，部分代码参考了 jittor-gan。