[资源]基于 Pytorch 的 TorchGAN开源了!
之前推荐过一个基于 TensorFlow 的 GAN 框架–谷歌开源的 GAN 库–TFGAN。
而最近也有一个新的 GAN 框架工具,并且是基于 Pytorch 实现的,项目地址如下:
https://github.com/torchgan/torchgan
对于习惯使用 Pytorch 框架的同学,现在可以采用这个开源项目快速搭建一个 GAN 网络模型了!
目前该开源项目有 400+ 星,它给出了安装的教程、API 文档以及使用教程,文档的地址如下:
https://torchgan.readthedocs.io/en/latest/
安装
对于 TorchGAN 的安装,官网给出 3 种方法,但实际上目前仅支持两种安装方式,分别是pip
方式安装以及源码安装,采用conda
安装的方法目前还不支持。
Pip 安装方法
安装最新的发布版本的命令如下:
$ pip3 install torchgan
而如果是最新版本:
$ pip3 install git+https://github.com/torchgan/torchgan.git
Conda 安装
这是目前版本还不支持的安装方式,将会在v0.1
版本实现这种安装方法。
源码方式安装
按照下列命令的顺序执行来进行从源码安装
$ git clone https://github.com/torchgan/torchgan
$ cd torchgan
$ python setup.py install
依赖库
必须按照的依赖库:
- Numpy
- Pytorch 0.4.1
- Torchvision
可选
- TensorboardX:主要是为了采用
Tensorboard
来观察和记录实验结果。安装通过命令pip install tensorboardX
- Visdom:为了采用
Xisdom
进行记录。安装通过命令pip install visdom
API 文档
API 的文档目录如下:
从目录主要分为以下几个大类:
- torchgan.layers:包含当前常用的用于构建 GAN 结构的一些网络层,包括残差块,Self-Attention,谱归一化(Spectral Normalization)等等
- torchgan.logging:提供了很强的可视化工具接口,包括对损失函数、梯度、测量标准以及生成图片的可视化等
- torchgan.losses:常见的训练 GANs 模型的损失函数,包括原始的对抗损失、最小二乘损失、WGAN的损失函数等;
- torchgan.metrics:主要是提供了不同的评判测量标准
- torchgan.models:包含常见的 GAN 网络结构,可以直接使用并且也可以进行拓展,包括 DCGAN、cGAN等
- torchgan.trainer:主要是提供训练模型的函数接口
教程
教程部分如下所示:
教程给出了几个例子,包括 DCGAN、Self-Attention GAN、CycleGAN 例子,以及如何自定义损伤的方法。
对于 Self-Attention GAN,还提供了一个在谷歌的 Colab 运行的例子,查看链接:
https://torchgan.readthedocs.io/en/latest/tutorials/sagan.html
小结
最后,再给出 Github 项目的链接和文档的对应链接地址:
https://github.com/torchgan/torchgan
https://torchgan.readthedocs.io/en/latest/index.html
欢迎关注我的微信公众号–机器学习与计算机视觉,或者扫描下方的二维码,大家一起交流,学习和进步!
之前分享的资源和教程文章有:
- 推荐几本数据结构算法书籍和课程
- [资源分享] Github上八千Star的深度学习500问教程
- [资源分享] 吴恩达最新《机器学习训练秘籍》中文版可以免费下载了!
- [资源分享] TensorFlow 官方中文版教程来了
- 必读的AI和深度学习博客
- [教程]一份简单易懂的 TensorFlow 教程
- 谷歌开源的 GAN 库–TFGAN
[资源]基于 Pytorch 的 TorchGAN开源了!相关推荐
- Kornia开源可微分计算机视觉库,基于Pytorch,GitHub 3000星
点击上方"3D视觉工坊",选择"星标" 干货第一时间送达 编辑丨机器之心 OpenCV 创始人 Gary Bradski 等人近期发表了一篇 Kornia 的综 ...
- FaceBook开源PyTorch3D:基于PyTorch的新3D计算机视觉库
关注上方"深度学习技术前沿",选择"星标公众号", 资源干货,第一时间送达! PyTorch 进军三维计算机视觉了,现在,你的神经网络可以更轻松地和 3D 数据 ...
- 艾伦人工智能研究院开源AllenNLP,基于PyTorch轻松构建NLP模型
艾伦人工智能研究院(AI2)开源AllenNLP,它是一个基于PyTorch的NLP研究库,利用深度学习来进行自然语言理解,通过处理低层次的细节.提供高质量的参考实现,能轻松快速地帮助研究员构建新的语 ...
- 清华大学开源迁移学习算法库:基于PyTorch实现,支持轻松调用已有算法
机器之心报道 编辑:魔王 作者:清华大学大数据研究中心 近日,清华大学大数据研究中心机器学习研究部开源了一个高效.简洁的迁移学习算法库 Transfer-Learn,并发布了第一个子库--深度领域自适 ...
- 【深度学习】基于Torch的Python开源机器学习库PyTorch卷积神经网络
[深度学习]基于Torch的Python开源机器学习库PyTorch卷积神经网络 文章目录 1 CNN概述 2 PyTorch实现步骤2.1 加载数据2.2 CNN模型2.3 训练2.4 可视化训练 ...
- 【深度学习】基于Torch的Python开源机器学习库PyTorch回归
[深度学习]基于Torch的Python开源机器学习库PyTorch回归 文章目录1 torch.autograd 2 torch.nn.functional 3 详细的回归DEMO3.1 DATAS ...
- 【深度学习】基于Torch的Python开源机器学习库PyTorch概述
[深度学习]基于Torch的Python开源机器学习库PyTorch概述 文章目录 1 PyTorch简介 2 环境搭建 3 Hello world!3.1 Tensors (张量)3.2 操作 4 ...
- 基于PyTorch的GAN框架TorchGAN:用架构级API轻松定制GAN项目
机器之心报道 参与:刘晓坤 TorchGAN 是基于 PyTorch 的 GAN 设计开发框架.该框架旨在为流行的 GAN 提供构造模块,且允许为前沿研究进行定制化. 使用 TorchGAN 的模块化 ...
- 轻量高效!清华智能计算实验室开源基于PyTorch的视频 (图片) 去模糊框架SimDeblur
作者丨科技猛兽 编辑丨极市平台 清华大学自动化系智能计算实验室团队开源基于 PyTorch 的视频 (图片) 去模糊框架 SimDeblur. 基于 PyTorch 的视频 (图片) 去模糊框架 Si ...
最新文章
- C++——运算符的重载---以成员函数方式重载---以友元函数方式重载
- ESXI开启snmp协议方法
- jquery插件 --- 图表 表格
- 对Unity的Resources目录进行改名
- C++CLR类库封装Native类库并用C#调用
- MediaRecorder录像怎么旋转呀?
- UIView的一些基本方法 init、loadView、viewDidLoad、viewDidUnload、dealloc
- html引入php文件中的函数,手把手教你在html中引入另一个html文件的方法(详解)...
- mysql 逆序排序_将一组乱序的字符进行排序进行升序和逆序输出
- Linux文件目录类指令
- 2021年中国再生纱市场趋势报告、技术动态创新及2027年市场预测
- 【Java篇】 Java小游戏中加入背景音乐
- Hexo 添加百度统计
- 74CMS 3.0 SQL注入漏洞前台
- 清朝十二帝记忆顺口溜
- wstmart 对目录进行授权
- IronPython团队宣言
- c语言与多字节编码,什么是单字节,双字节和多字节编码
- 【Prometheus 自主无人机平台 】无人机蜂群拦截系统作战构想与关键技术论文解读
- 【论文阅读】Efficient Illuminant Estimation for Color Constancy Using Grey Pixels