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轮训练,解压后放在代码文件同目录即可运行,否则程序将自动进行训练.
CGAN-Jittor
本项目借助Jittor框架实现了一个简易的CGAN,目前能够做到生成0-9共十个手写数字,并按0-9的顺序显示为图片
主要运行环境
python 3.8,Jittor 1.3.7.13
运行方法
项目包含一个文件,即main.py,main.py在训练模型之后直接进行测试,可生成0-9的图片并组合输出,默认输出为result.png.
CGAN算法说明
CGAN(条件生成对抗网络)由生成器和辨别器构成,其中生成器接受随机噪声和指定标签生成图片,辨别器接受真实图片和生成图片并判断图片是否为真,两者互相对抗进行训练。本项目实现的CGAN使用mnist手写数字数据集作为真实图片训练,因此仅能产生0-9的数字。
数据集及模型说明
数据集自行下载即可,解压后放在mnist_data/目录下,数据集链接:数据集 模型文件链接:模型文件,解压后放在saved_model/下即可 模型已经过400轮训练,解压后放在代码文件同目录即可运行,否则程序将自动进行训练.