jittor-TrickySightseeing-LandscapeImageGeneration

Jittor 草图生成风景比赛

简介

本项目包含了TrickySightseeing小队的草图生成风景比赛的代码实现。

安装

本项目可在 1 张 A100 上运行，batch_size=10。

运行环境

• ubuntu 20.04 LTS
• python >= 3.7
• jittor >= 1.3.0 其它详见requirements.txt

训练（包含推理）

单卡训练可运行以下命令：

bash scripts/train.sh

在每一个epoch训完后会进行测试集推理。

推理

单卡推理可运行以下命令：

bash scripts/test.sh

即可load位于’results/saved_models/‘中的预训练模型进行测试，并将测试结果保存至’results/‘