基于GauGAN的风景图片生成器_补充仓库

简介

本项目包含了第二届计图挑战赛中Benchmark小组对赛道一风景生成比赛的代码实现。

本仓库中实现的代码基于pytorch实现

在原本GauGAN改进算法的基础上，添加了VAN(Visual Attention Network)与信息检索的内容，成为全新的网络

但是由于算力不足，还没有完整运行，未能提交测评

配置环境

本项目可在 3090 上运行，训练时间约为 4 天。

运行环境

• ubuntu 20.04 LTS
• python >= 3.8

安装依赖

执行以下命令安装 python 依赖

pip install -r requirements.txt

训练

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

python main.py