目录
目录README.md

cgan_jittor

一个用 Jittor 框架实现的针对手写数字生成的 Conditional GAN 模型,同时也是计图 (Jittor) 人工智能算法挑战赛的热身赛题。

简述

本项目的应用

本项目调用 Jittor 框架,训练一个 Conditional GAN 手写数字生成模型,可以生成任意数字序列的手写数字图像,

为何使用 Jittor

Jittor(计图)是清华大学发布的基于动态编译、使用创新的元算子与统一计算图的自研深度学习框架。Jittor 易于使用,且提供高性能的优化,达到比 PyTorchTensorFlow 更加强劲的训练性能。

关于 Conditional GAN

Conditional GAN 是广为人知的生成对抗网络(Generative Adversarial Network, GAN)的一个变种。它通过在生成器(Generator)和判别器(Discriminator)中添加额外信息,以为生成的图像提供必要的条件控制。在手写数字生成任务中,这种额外信息可以选用数字 0~9 的类别标签。

安装与使用

本项目无需安装。克隆本项目后,在 CGAN.py 所在路径下运行 python CGAN.py,即可开始训练手写数字生成模型,最终的生成结果为“0123456789”的手写数字图像,将保存在同级目录下的 result.png 中。可以修改 CGAN.py 第 204 行的 number 字符串,以生成定制的手写数字。

友情链接

计图 (Jittor) 官网 计图 (Jittor) 人工智能算法挑战赛

关于

A Jittor implementation of Conditional GAN (CGAN).

34.0 KB
邀请码
    Gitlink(确实开源)
  • 加入我们
  • 官网邮箱:gitlink@ccf.org.cn
  • QQ群
  • QQ群
  • 公众号
  • 公众号

©Copyright 2023 CCF 开源发展委员会
Powered by Trustie& IntelliDE 京ICP备13000930号