CGAN-Jittor

项目简介

本项目为第五届计图人工智能挑战赛的热身赛项目，基于 jittor 深度学习框架，在数字图片数据集 MNIST 上训练 Conditional GAN（CGAN）模型，生成特定数字的图像

安装要求

环境依赖

# Python版本要求
Python >= 3.7

# 主要依赖包
pip install jittor
pip install numpy
pip install pillow
pip install argparse

使用方法

快速开始

1. 克隆项目

   git clone git@code.gitlink.org.cn:HaozhiZheng/CGAN_jittor.git
   cd CGAN_jittor

2. 运行训练

   python CGAN.py

3. 自定义参数训练

   python CGAN.py --n_epochs 200 --batch_size 128 --lr 0.0001

主要参数说明

参数	默认值	说明
--n_epochs	100	训练轮数
--batch_size	64	批次大小
--lr	0.0002	学习率
--latent_dim	100	噪声向量维度
--img_size	32	生成图片尺寸
--sample_interval	1000	图片采样间隔

友情链接

计图（Jittor）深度学习框架：Jittor官方文档

第五届计图人工智能挑战赛：第五届计图人工智能挑战赛