Jittor 计图挑战热身赛 baseline

Jittor计图挑战热身赛代码仓库。队伍：YCLab

简介

本项目包含了第三届计图挑战赛计图 - 草图生成风景比赛的代码实现。本项目的特点是：采用了 CGAN方法，完成了样例代码的TODO部分，成功生成了队员的手机号。

安装

本项目可在 1 张 2060s 上运行，训练时间约为 6 小时。

运行环境

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

安装依赖

• numpy >= 1.20

预训练模型

discriminator_last.pkl与generator_last.pkl是预训练好的模型，须与CGAN.py置于同一目录下。

训练

训练可以执行如下指令：

python CGAN.py

训练完成之后便会在目录下得到result.png

推理

热身赛暂时未实现评估程序。