jittor 第四届计图人工智能挑战赛-热身赛 CGAN

简介

本项目包含了第四届计图挑战赛。基于国产框架jittor(计图)实现了CGAN网络。 本项目的特点是通过输入信息，可以生成高质量的特定数字图像。

安装

本项目在V100上运行15分钟左右即可训练出一个较为可观的CGAN模型。结合计图框架，对显卡要求较低RTX20系列即可使用。 #### 运行环境

• Ubuntu 20.04.6 LTS
• Python >= 3.7
• Jittor >= 1.3.0 (安装教程：https://cg.cs.tsinghua.edu.cn/jittor/download/)
• CUDA Version: 12.0

  安装依赖

  pip install -r requirements.txt

  致谢

  | 对参考的开源库、官方予以致谢 此项目基于比赛官方(https://www.educoder.net/competitions/Jittor-5)以及国产框架jittor得以实现，部分代码参考了[jittor-gan](https://github.com/Jittor/gan-jittor)