第三届计图人工智能挑战赛：风格及语义引导的风景图像生成赛题（赛题一）baseline

简介

本项目基于比赛baseline以及参考了SPADE原论文的官方开源代码：SPADE。在此基础上进行了进一步训练以及调参。 测试时，SPADE网络可以完成一张参考的ref图+一张label Mask图通过网络输出一张生成图。 此外，仓库还实现了SEAN，OASIS两个算法的jittor版本。 本仓库由清华大学计算机系熊方舟，彭梓健，胡越舟维护。

运行环境

ubuntu 20.04 LTS python >= 3.7 jittor >= 1.3.0

安装依赖

执行以下命令安装 python 依赖

pip install -r requirements.txt

训练

在单卡上训练：

sh train.sh

此前需要修改train.sh，其内容为：

# train.sh
CUDA_VISIBLE_DEVICES="0" python train.py --input_path {训练数据集路径（即train_resized文件夹所在路径）}

测试

在单卡上进行测试：

sh test.sh

此前需要：

1. 将label与img的映射关系（label_to_img.json）放置在gaugan目录下
2. 修改test.sh，其内容为： ```bash CUDA_VISIBLE_DEVICES=”0” python test.py \

–input_path {测试数据集路径（即labels文件夹所在路径），它提供label mask图}
–img_path {训练数据集的图片路径（即train_resized/imgs文件夹所在路径，它提供ref图）} –which_epoch {使用的模型的epoch数目}

致谢

原作者将论文的 pytorch 版本的源代码，迁移到了 Jittor 框架当中。其中借鉴了开源社区 Spectral Normalization 的代码，以及重度参考了原论文的官方开源代码： SPADE; SEAN; OASIS。