Github|基于 Jittor 的 GAN 模型库

Posted 2021-04-10 算法猿的成长

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

前言

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 项目。

---

简介

我们的 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 小时。

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

---

模型使用样例

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

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

---

---

精选AI文章

1.  

2. 

3. 

4. 

5. 

6. 

7. 

8. 

精选python文章

1.  

2. 

3. 

4. 

5. 

精选教程资源文章

1. 

2. 

3. 

4. 

5. 

 

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