目录
目录README.md

CGAN-Jittor

本项目借助Jittor框架实现了一个简易的CGAN,目前能够做到生成0-9共十个手写数字,并按0-9的顺序显示为图片

主要运行环境

python 3.8,Jittor 1.3.7.13

运行方法

项目包含一个文件,即main.py,main.py在训练模型之后直接进行测试,可生成0-9的图片并组合输出,默认输出为result.png.

注:程序已经支持cuda加速

CGAN算法说明

CGAN(条件生成对抗网络)由生成器和辨别器构成,其中生成器接受随机噪声和指定标签生成图片,辨别器接受真实图片和生成图片并判断图片是否为真,两者互相对抗进行训练。本项目实现的CGAN使用mnist手写数字数据集作为真实图片训练,因此仅能产生0-9的数字。

数据集及模型说明

数据集自行下载即可,解压后放在mnist_data/目录下,数据集链接:数据集 模型文件链接:模型文件,解压后放在saved_model/下即可 模型已经过400轮训练,解压后放在代码文件同目录即可运行,否则程序将自动进行训练.

关于
42.0 KB
邀请码