Jittor实现CGAN模型 计图热身赛

第二届计图人工智能挑战赛 热身赛 手写数字生成

简介

• 本项目是在“头歌”平台上参加“第二届计图人工智能挑战赛 ”——热身赛的项目仓库
• 深度学习框架来源于Jittor,实现了CGAN模型
• 代码框架来自于https://github.com/Jittor/gan-jittor/blob/master/ competition/warm_up_comp/CGAN.py 对TODO部分做了修改

安装项目

• 检查python版本 >=3.8 Jittor框架安装详见 https://cg.cs.tsinghua.edu.cn/jittor/download/
• 代码部分下载即可
• *.pkl 为一些模型数据

数据集

代码中自动下载，无需另外下载。

使用

(windows) python CGAN.py

(Ubuntu)  python3 CGAN.py