JGAN/competition/landscape_comp
zhouwy19 6280b17ed3 ignore running_mean/var warning 2022-05-06 20:30:00 +08:00
..
scripts add competition 2022-04-19 10:33:09 +08:00
datasets.py add competition 2022-04-19 10:33:09 +08:00
models.py ignore running_mean/var warning 2022-05-06 20:30:00 +08:00
pix2pix.py Update pix2pix.py 2022-04-25 18:47:56 +08:00
readme.md add competition 2022-04-19 10:33:09 +08:00

readme.md

Jittor 第二届草图生成风景比赛 baseline

Requirements

jittor
pillow
opencv-python

Train

单卡训练,需要修改脚本里的数据路径

bash scripts/single_gpu.sh

多卡训练,需要修改脚本里的数据路径

bash scripts/multi_gpu.sh

代码中注释掉了eval的部分等到测试数据发布之后您可以取消注释进行评测。也可在训练阶段自动分配一部分数据集为测试集进行训练。