Jittor 手写数字生成赛 CGAN
程序运行结果:

简介
本项目包含了第二届计图挑战赛计图 - 手写数字生成赛的代码实现。本项目的特点是:采用了CGAN方法对MNIST数据集处理,取得了生成指定手写数字图片的效果。
安装
python -m pip install jittor
测试是否可用
python -m jittor.test.test_core
python -m jittor.test.test_example
本项目可在 1 张 RTX4000 上运行。
运行环境
- windows 11
- python >= 3.8
- jittor >= 1.3.0
安装依赖
执行以下命令安装 python 依赖
pip install -r requirements.txt
预训练模型
无
数据预处理
使用Jittor内嵌的MNIST数据集,无需下载预处理,直接程序加载。
训练
python CGAN.py
推理
无
致谢
此项目基于论文 A Style-Based Generator Architecture for Generative Adversarial Networks 实现,部分代码参考了 jittor-gan。
注意事项
无
Jittor 手写数字生成赛 CGAN
程序运行结果:
简介
本项目包含了第二届计图挑战赛计图 - 手写数字生成赛的代码实现。本项目的特点是:采用了CGAN方法对MNIST数据集处理,取得了生成指定手写数字图片的效果。
安装
本项目可在 1 张 RTX4000 上运行。
运行环境
安装依赖
执行以下命令安装 python 依赖
预训练模型
无
数据预处理
使用Jittor内嵌的MNIST数据集,无需下载预处理,直接程序加载。
训练
推理
无
致谢
此项目基于论文 A Style-Based Generator Architecture for Generative Adversarial Networks 实现,部分代码参考了 jittor-gan。
注意事项
无