Jittor 第二届计图人工智能挑战赛 热身赛方案
战队名:我和我的两个可爱师叔

简介
本项目包含了第二届计图挑战赛计图 - 热身赛的代码实现。主要方法基于CGAN。
安装
本项目可在 1 张 RTX 3090 上运行,训练时间约为30分钟。
运行环境
- ubuntu 20.04 LTS
- python >= 3.7
- jittor >= 1.3.0
安装依赖
执行以下命令安装 jittor
pip install jittor
预训练模型
预训练模型模型下载地址为 百度网盘,提取码:76j4。下载后放在当前目录即可。
训练
模型的训练可以运行以下命令:
python CGAN.py
训练完成后会根据设置的手机号自动进行推理。
若用提供的预训练模型直接测试,注释掉CGAN.py中的训练部分即可。
致谢
此项目基于论文 Conditional Generative Adversarial Nets 实现,部分代码参考了 jittor-gan。
Jittor 第二届计图人工智能挑战赛 热身赛方案
战队名:我和我的两个可爱师叔
简介
本项目包含了第二届计图挑战赛计图 - 热身赛的代码实现。主要方法基于CGAN。
安装
本项目可在 1 张 RTX 3090 上运行,训练时间约为30分钟。
运行环境
安装依赖
执行以下命令安装 jittor
预训练模型
预训练模型模型下载地址为 百度网盘,提取码:76j4。下载后放在当前目录即可。
训练
模型的训练可以运行以下命令:
训练完成后会根据设置的手机号自动进行推理。 若用提供的预训练模型直接测试,注释掉CGAN.py中的训练部分即可。
致谢
此项目基于论文 Conditional Generative Adversarial Nets 实现,部分代码参考了 jittor-gan。