CGAN_jittor
简介:
A Jittor implementation of Conditional GAN (CGAN).
本题目为第三届计图人工智能挑战赛计图挑战热身赛,南开大学计算机图形学小作业。
技术介绍
比赛主页链接如下:
第三届计图人工智能挑战赛 (educoder.net)
在数字图片数据集 MNIST 上训练 Conditional GAN(Conditional generative adversarial nets)模型,通过输入一个随机向量 z 和额外的辅助信息 y (如类别标签),生成特定数字的图像。
其中使用到了计图(Jittor),安装与使用链接如下:
https://cg.cs.tsinghua.edu.cn/jittor/resources/
Conditional GAN是由 Ian Goodfe等人提出的一个GAN的改进模型,在GAN的基础上加入用户指定的标签,生成器根据标签,生成与标签相关的样本;判别器根据标签判断图片(在图片真实的基础上加入与标签相关的要求,只有二者同时满足才标记为1,其余情况标记为0)。从而达到根据指定标签生成图片的功能,论文链接如下:
https://arxiv.org/pdf/1411.1784.pdf%EF%BC%88CGAN%EF%BC%89
环境要求
主要为Jittor框架的环境要求:(来自赛题官网)
Jittor框架对环境要求如下:
- 操作系统: Ubuntu >= 16.04 或 Windows Subsystem of Linux(WSL)
- Python:版本 >= 3.7
- C++编译器 (需要下列至少一个)
- g++ (>=5.4.0)
- clang (>=8.0)
- GPU 编译器(可选):nvcc >=10.0
- GPU 加速库(可选):cudnn-dev (cudnn开发版, 推荐使用tar安装方法,参考链接)
如果您不希望手动配置环境,我们推荐使用 Docker 进行安装。 除此之外,您还可以使用 pip 安装和手动安装。
注意:目前Jittor通过WSL的方式在Windows操作系统上运行,WSL的安装方法请参考微软官网,WSL版本目前尚不支持CUDA。
结果展示
本代码在给定的框架基础上进行修改,生成图像见reault.png
MIT License
Copyright (c) [2023] [丁延峰]
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the “Software”), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:
The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.
CGAN_jittor
简介:
A Jittor implementation of Conditional GAN (CGAN).
本题目为第三届计图人工智能挑战赛计图挑战热身赛,南开大学计算机图形学小作业。
技术介绍
比赛主页链接如下:
第三届计图人工智能挑战赛 (educoder.net)
在数字图片数据集 MNIST 上训练 Conditional GAN(Conditional generative adversarial nets)模型,通过输入一个随机向量 z 和额外的辅助信息 y (如类别标签),生成特定数字的图像。
其中使用到了计图(Jittor),安装与使用链接如下:
https://cg.cs.tsinghua.edu.cn/jittor/resources/
Conditional GAN是由 Ian Goodfe等人提出的一个GAN的改进模型,在GAN的基础上加入用户指定的标签,生成器根据标签,生成与标签相关的样本;判别器根据标签判断图片(在图片真实的基础上加入与标签相关的要求,只有二者同时满足才标记为1,其余情况标记为0)。从而达到根据指定标签生成图片的功能,论文链接如下:
https://arxiv.org/pdf/1411.1784.pdf%EF%BC%88CGAN%EF%BC%89
环境要求
主要为Jittor框架的环境要求:(来自赛题官网)
Jittor框架对环境要求如下:
如果您不希望手动配置环境,我们推荐使用 Docker 进行安装。 除此之外,您还可以使用 pip 安装和手动安装。
注意:目前Jittor通过WSL的方式在Windows操作系统上运行,WSL的安装方法请参考微软官网,WSL版本目前尚不支持CUDA。
结果展示
本代码在给定的框架基础上进行修改,生成图像见reault.png
MIT License
Copyright (c) [2023] [丁延峰]
Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the “Software”), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.