Jittor 基于ConditionalGAN的手写数字生成

最终结果

简介

本项目包含了第二届计图挑战赛计图 - 热身赛 - 手写数字生成的代码实现。基于Conditional GAN将随机噪声和类别标签映射为数字图片。

安装

本项目可在 2 张 3090上运行，训练时间约为1 小时。

运行环境

• ubuntu 20.04 LTS
• python >= 3.7
• jittor >= 1.3.0

安装依赖

遵循Jittor官方仓库指导

https://github.com/Jittor/gan-jittor

预训练模型

预训练模型模型下载地址为 https://github.com/Jittor/gan-jittor

训练处理

此项目基于论文 A Style-Based Generator Architecture for Generative Adversarial Networks 实现，部分代码参考了 jittor-gan。