Jittor CGAN 计图热身挑战赛

简介

| 简单介绍项目背景、项目特点

本项目包含了第二届计图Jittor挑战赛计图 - 计图挑战热身赛的代码实现。本项目目标为生成指定数字序列的图片，采用了CGAN方法在MNIST数据集上训练，最终测评分数0.9996。

安装

sudo apt install python3.9-dev libomp-dev
python3.9 -m pip install jittor
python3.9 -m jittor.test.test_example

运行环境

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

安装依赖

执行以下命令安装 python 依赖

pip install -r requirements.txt

预训练模型

无

数据预处理

使用Jittor中的的MNIST数据集。

训练

python CGAN.py

推理

直接执行python CGAN.py就会在同个目录下生成result.png

致谢

| 对参考的论文、开源库予以致谢，可选

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