Jittor 生成手写数字比赛
简介
本项目包含了第二届计图挑战赛热身赛手写数字生成的代码实现。
本项目的特点是:采用了CGAN方法对MINST手写数字数据集处理,成功生成了手机号码的手写数字。
运行环境
- ubuntu 20.04 LTS
- python >= 3.7
- jittor >= 1.3.0
安装
安装好对应python版本后,可通过运行以下代码安装jittor
pip install jittor
通过运行Resnet18训练测试来检查Jittor的完整性
python3.7 -m jittor.test.test_resnet
训练
生成测试集上的结果可以运行以下命令:
python CGAN.py
致谢
此项目基于论文 A Style-Based Generator Architecture for Generative Adversarial Networks 实现,部分代码参考了 jittor-gan。
Jittor 生成手写数字比赛
简介
本项目包含了第二届计图挑战赛热身赛手写数字生成的代码实现。 本项目的特点是:采用了CGAN方法对MINST手写数字数据集处理,成功生成了手机号码的手写数字。
运行环境
安装
安装好对应python版本后,可通过运行以下代码安装jittor
通过运行Resnet18训练测试来检查Jittor的完整性
训练
生成测试集上的结果可以运行以下命令:
致谢
此项目基于论文 A Style-Based Generator Architecture for Generative Adversarial Networks 实现,部分代码参考了 jittor-gan。