目录
目录README.md

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-第二届计图人工智能挑战赛热身赛

34.0 KB
邀请码