目录
目录README.md

CGAN-Jittor

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

代码框架来源于头歌第三届计图人工智能挑战赛热身赛部分https://www.educoder.net/competitions/index/Jittor-4

主要运行环境

python 3.8,Jittor 1.3.7.13

运行方法

项目仅包含一个文件,即CGAN.py,运行后输入要生成的数字串,即可在目录下输出对应生成图片,图片名默认为result.png.

注:程序已经支持cuda加速

CGAN算法说明

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

数据集及模型说明

数据集运行代码即可由Jittor自动获取,模型文件链接:https://pan.baidu.com/s/1ac95HJ3fHmi9ECvZapvS8Q?pwd=1234 模型已经过100轮训练,解压后放在代码文件同目录即可运行,否则程序将自动进行训练.

关于

A Jittor implementation of Conditional GAN(CGAN).

35.0 KB
邀请码