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上有哪些比较好的计算机视觉/机器视觉的项目?
欢迎关注我的微信公众号--算法猿的成长,或者扫描下方的二维码,大家一起交流,学习和进步!
如果觉得不错,在看、转发就是对小编的一个支持!
Github|基于 Jittor 的 GAN 模型库相关推荐
- 基于Jittor框架实现LSGAN图像生成对抗网络
基于Jittor框架实现LSGAN图像生成对抗网络 生成对抗网络(GAN, Generative Adversarial Networks )是一种深度学习模型,是近年来复杂分布上无监督学习最具前景的 ...
- 基于PyTorch的GAN框架TorchGAN:用架构级API轻松定制GAN项目
机器之心报道 参与:刘晓坤 TorchGAN 是基于 PyTorch 的 GAN 设计开发框架.该框架旨在为流行的 GAN 提供构造模块,且允许为前沿研究进行定制化. 使用 TorchGAN 的模块化 ...
- Transformer也能生成图像,新型ViTGAN性能比肩基于CNN的GAN
©作者 | Panda 来源 | 机器之心 Transformer 已经为多种自然语言任务带来了突飞猛进的进步,并且最近也已经开始向计算机视觉领域渗透,开始在一些之前由 CNN 主导的任务上暂露头角. ...
- 【图像超分辨率】基于ResNet或GAN的遥感图像超分辨率论文
基于ResNet或GAN的遥感图像超分辨率论文 <空间感知残差网络的遥感图像超分辨率重建> 操作: 遥感图像特点: 网络模型: 去掉批处理层的原因: 具体操作 损失方程: <改进的残 ...
- [论文翻译]基于图像自适应GAN重建
基于图像自适应GAN重构 摘要 介绍 相关工作 提出的方法 Image-Adaptive图像自适应方法 IAGAN的数学原理 对观察结果的"硬"和"软"遵从性 ...
- DL之GAN:HighNewTech基于计算机视觉领域GAN技术—最新黑科技之秒变宝宝——回到最初的样子
DL之GAN:High&NewTech基于计算机视觉领域GAN技术-最新黑科技之秒变宝宝--回到最初的样子 导读 8月14日,快手APP上线名为"变小孩"的魔法表 ...
- Github项目推荐|可视化 GAN 的训练过程
点击上方"算法猿的成长",关注公众号,选择加"星标"或"置顶" 总第 137 篇文章,本文大约 1000 字,阅读大约需要 5 分钟 今天介 ...
- Github | 基于Pytorch可微分计算机视觉库
点上方蓝字计算机视觉联盟获取更多干货 在右上方 ··· 设为星标 ★,与你不见不散 仅作学术分享,不代表本公众号立场,侵权联系删除 转载于:机器之心 AI博士笔记系列推荐 周志华<机器学习> ...
- 基于MNIST实现GAN(pytorch)
基于MNIST实现生成对抗网络(pytorch逐行实现) 本文是pytorch逐行实现GAN网络,作为一个基础GAN框架来学习,以后编写复杂的GAN的衍生网络框架都是同样的思想 import nump ...
最新文章
- mysql windows 管道连接,科技常识:Windows Server 2016 MySQL数据库安装配置详细安装教程...
- 常用AT指令集 (转)
- oracle中nvarchar2字符集不匹配
- rust种的南瓜为什么老是消失_科技的力量!3种“奇葩”的发明,你都见过吗?...
- python大佬养成计划--协程实现TCP连接
- python语言中、外部模块先导入、再使用_python引入导入自定义模块和外部文件--转载Sumomo的博客...
- Convert Sorted List to Binary Search Tree ------C++ 递归创建平衡二叉查找树
- MySQL的chap服务器_路由器chap的双向配置命令
- kong自定义插件(修改官方插件)
- psim扰动观察法编程c语言,一种数字PID控制扰动观察法光伏电池MPPT仿真.doc
- 华为光猫查看LOID密码
- 角度传感器测量实验 GY-25
- 3.1 视频服务器介绍
- UTM(Urchin Tracking Module)简介
- Artificial Neural Networks FileStorage of OpenCV
- Android手机上,利用bat脚本模拟用户操作
- linux的第一个内核,Linus Torvalds启动了第一个RC版本的Linux内核4.15
- php两个并排的按钮,如何并排设置2个按钮
- 你知道数据运营日常主要工作吗?
- 基于JSP技术的游泳馆管理系统
热门文章
- matlab 万能,matlab 万能实用的线性曲线拟合方法
- php splqueue 5.5安装,解析PHP标准库SPL数据结构
- 50欧姆线设计 高频pcb_硬件设计基础100问(三)
- linux bind命令,LINUX命令bind-系统管理-显示或设置键盘按键与其相关的功能
- MSP430F5529 DriverLib 库函数学习笔记(二)GPIO
- 输入框限制只能输入数字,正数、负数、0,最多两位小数;数字输入框可以输入负数,并最多保留两位小数;el-number-input去掉四舍五入和自动补齐小数;
- Qt下Undefined reference to 'vtable for xxx'
- [react] react的性能优化在哪个生命周期?它优化的原理是什么?
- [react] React中getInitialState方法的作用是什么
- 学会用taro封装一个组件