目录
目录README.md

第二届计图人工智能挑战赛热身赛

简介

本项目为第二届计图人工智能挑战赛-热身赛的代码实现。本项目基于示例代码框架补充完成,可完成热身赛的要求(生成注册绑定的手机号图片)。

安装

本项目可在 1 张 3090 上运行,训练时间约为 1 小时。

运行环境

  • ubuntu 20.04 LTS
  • python >= 3.8
  • jittor >= 1.3.3

安装依赖

执行以下命令安装 python 依赖

pip install -r requirements.txt

训练&生成

训练(单卡)以及生成 result.png 可运行以下命令:

python CGAN.py --output_number [target_number]

致谢

本项目代码框架来自于 gan-jittor/competition/warm_up_comp

关于

A Jittor implementation of Conditional GAN (CGAN).

29.0 KB
邀请码