Github|基于 Jittor 的 GAN 模型库

点击上方“算法猿的成长“，关注公众号，选择加“星标“或“置顶”

总第 142 篇文章，本文大约 1300 字，阅读大约需要 5 分钟

前言

今天介绍一个 Github 项目---集成了 27 种 GAN 模型的 Jittor-GAN，项目地址：

https://github.com/Jittor/gan-jittor?u=2181051220&m=4512179795555993&cu=2181051220&ru=1402400261&rm=4512157284683484

包含了从 2014 年最开始的 GAN 算法模型代码，到 2019 年的的一个 GAN 模型代码，其中有非常著名的 Pix2Pix，CycleGAN，StarGAN 等，也有比较冷门点的 GAN 模型算法。

下面就简单介绍一下这个 Github 项目。

---

简介

本项目的代码基于 Pytorch-GAN，项目地址：https://github.com/eriklindernoren/PyTorch-GAN

我们的 GAN模型库--Jittor-GAN 支持 27 种 GAN 模型，下面这份表格是来自谷歌学术的最新引用情况。GAN 是从 2014 年提出，然后产生了很多优秀的 GAN 的改进工作。这 27 种 GAN 算法总共被引用了 60953 次，平均每篇论文被引用 2176 次。

我们还对比了基于 Jittor 的 GAN 模型库和 Pytorch 的性能，结果如下所示，图中的数字表示 Jittor 实现的模型相对于 Pytorch 的速度提升情况，最快的情况是提升了 283%，平均提升速度也是 185%。

在另一个展示角度，假设 Pytorch 需要的训练时间是 100 个小时，而 Jittor 实现的 GAN 模型代码训练时间会少很多，最快的 GAN 模型只需要 35 个小时，平均需要 57 小时。

另外，代码里使用到的 Jittor，其官网地址：

http://cg.cs.tsinghua.edu.cn/jittor/

官网对 Jittor 的介绍：

Jittor 是一个基于即时编译和元算子的高性能深度学习框架，整个框架在即时编译的同时，还集成了强大的Op编译器和调优器，为您的模型生成定制化的高性能代码。

Jittor前端语言为Python。前端使用了模块化的设计，类似于PyTorch，Keras，后端则使用高性能语言编写，如CUDA，C++。

---

安装

安装的命令如下：

$ git clone https://github.com/Jittor/gan-jittor.git
$ cd gan-jittor/
$ sudo python3.7 -m pip install -r requirements.txt

---

模型使用样例

这里简单举个例子来介绍如何运行这些 GAN 算法，这里以 CycleGAN 为例子，在 Jittor-GAN 的代码实现地址：

https://github.com/Jittor/gan-jittor/blob/master/models/cyclegan/cyclegan.py

运行代码例子如下，首先是下载对应的数据集，然后找到对应的文件夹，运行相应的代码

$ cd data/
$ bash download_cyclegan_dataset.sh monet2photo
$ cd ../models/cyclegan/
$ python3.7 cyclegan.py --dataset_name monet2photo

其他模型的使用例子也是如上所示

---

最后，该项目地址可以直接点击文章末尾底部”阅读原文“，或者也可以后台回复【jittor_gan】，获取项目代码的网盘链接。

---

精选AI文章

1.  2020年计算机视觉学习指南

2. 是选择Keras还是PyTorch开始你的深度学习之旅呢？

3. 编写高效的PyTorch代码技巧（上）

4. 编写高效的PyTorch代码技巧（下）

5. 深度学习算法简要综述(上)

6. 深度学习算法简要综述(下)

7. 10个实用的机器学习建议

8. 实战|手把手教你训练一个基于Keras的多标签图像分类器

精选python文章

1.  python数据模型

2. python版代码整洁之道

3. 快速入门 Jupyter notebook

4. Jupyter 进阶教程

5. 10个高效的pandas技巧

精选教程资源文章

1. [资源分享] TensorFlow 官方中文版教程来了

2. [资源]推荐一些Python书籍和教程，入门和进阶的都有！

3. [Github项目推荐] 推荐三个助你更好利用Github的工具

4. Github上的各大高校资料以及国外公开课视频

5. GitHub上有哪些比较好的计算机视觉/机器视觉的项目？

欢迎关注我的微信公众号--算法猿的成长，或者扫描下方的二维码，大家一起交流，学习和进步！

 

如果觉得不错，在看、转发就是对小编的一个支持！