Jittor 计图挑战热身赛 cGAN

战队名：海

简介

本项目包含了第二届计图挑战赛计图 - 热身赛的代码实现。主要方法基于CGAN。

安装

本项目可在 1 张 TITAN RTX 上运行，训练时间约为30分钟。

运行环境

ubuntu 20.04 LTS
python >= 3.7
jittor >= 1.3.0

安装说明

执行以下命令安装 jittor

pip install jittor

运行命令

模型的训练可以运行以下命令：

python CGAN.py

训练完成后会根据设置的手机号自动进行推理。

致谢

此项目大部分代码参考了 jittor-gan。