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

简介

本项目包含了第二届计图挑战赛计图 - 热身比赛-数字生成的代码实现。 本项目采用多类别的生成器和判别器，生成器和判别器的最后一层生成/鉴别每个特定类别的图像，其他部分和baseline完全相同。 本项目的测评分数为0.998.

安装

| 介绍基本的硬件需求、运行环境、依赖安装方法

运行环境

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

安装依赖

执行以下命令安装 python 依赖

pip install -r requirements.txt

jittor的安装方法请参考 https://cg.cs.tsinghua.edu.cn/jittor/download/

训练

训练可运行以下命令，数据集将被自动下载：

python CGAN.py --mode train

推理

生成测试集上的结果可以运行以下命令：

python CGAN.py --mode sample