目录
目录README.md

CGAN_Jittor

介绍

本项目是 第三届计图人工智能挑战赛 热身赛部分的赛题的一份解答。

本项目使用 Jittor 深度学习框架,在数字图片数据集 MNIST 上训练一个将随机噪声和类别标签映射为数字图片的 Conditional GAN 模型,生成指定数字序列对应的图片。

使用方式

安装

本项目使用 Jittor 框架,Jittor 框架目前支持 Linux 或 Windows(包括 WSL)。需要使用 Python 及 C++ 编译器(g++ 或 clang)。

Jittor 提供了三种安装方法:docker,pip 和手动安装,具体安装教程请参考:https://cg.cs.tsinghua.edu.cn/jittor/download/

运行

运行以下命令,即可进行模型训练,并生成手写数字图片。

python CGAN.py

训练结果 Demo

训练结果 Demo 如下,生成 13666671365542 的手写数字图片。

中间文件与结果

在训练过程中的中间文件与训练结果参见 清华云盘链接

关于

A Jittor implementation of Conditional GAN (CGAN).

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

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