专栏首页机器之心基于PyTorch的GAN框架TorchGAN:用架构级API轻松定制GAN项目

基于PyTorch的GAN框架TorchGAN:用架构级API轻松定制GAN项目

机器之心报道

参与:刘晓坤

TorchGAN 是基于 PyTorchGAN 设计开发框架。该框架旨在为流行的 GAN 提供构造模块,且允许为前沿研究进行定制化。

使用 TorchGAN 的模块化结构可以:

  • 在数据集上尝试流行的 GAN 模型;
  • 对传统损失函数、架构等插入新的损失函数、新的架构等;
  • 用多种 logging 后端无缝可视化 GAN 训练过程。

项目地址:https://github.com/torchgan/torchgan

TorchGAN 包由各种生成对抗网络和实用程序组成,这些网络和实用程序在训练时非常有用。该软件包提供了易于使用的 API,可用于训练流行的 GAN 以及开发 GAN 变体。作者写了一个教程文档,帮助你使用该软件包。

文档地址:https://torchgan.readthedocs.io/en/latest/

文档目录

TorchGAN 的文档包含入门、API 文档和教程三大部分。

入门部分会介绍 TorchGAN 的多种安装方法、必要的依赖包安装以及给该项目贡献的指南。

API 文档会介绍构建 GAN 网络的各种常用模块或层、损失函数、评估指标、经典模型(例如 InfoGAN、DCGAN 等)以及训练器,这些 API 可以帮助你快速从架构层面自定制模型。

教程部分会介绍具体的 GAN 项目构建过程,涉及了数据集构建、架构设计、超参数和优化器设置、损失函数定义、可视化、训练等机器学习项目实现的完整过程。作者使用 SAGAN 和 CycleGAN 作为典型案例的展示,还特地写了一个章节介绍如何自定制损失函数,这其实也是整个项目中最重要的环节之一,模型的训练是否收敛、收敛速度如何以及最终的收敛效果都受到损失函数定义的很大影响。

SAGAN 教程示例

本文分享自微信公众号 - 机器之心(almosthuman2014)

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

原始发表时间:2018-12-25

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 综述 | 一文帮你发现各种出色的GAN变体

    选自Guimperarnaublog 作者:Guim Perarnau 机器之心编译 参与:赵华龙、黄小天 你是否曾经想了解生成对抗网络(GAN)?也许你只是想...

    机器之心
  • 琼恩·雪诺和龙母的孩子会长啥样?让StyleGAN告诉你

    你有没有好奇过自己喜欢的电影或电视剧里的人物性别变换后是长啥样的?比如说,下面这位?

    机器之心
  • 这份攻略帮你「稳住」反复无常的 GAN

    生成对抗网络(GAN)是一类非常强大的神经网络,具有非常广阔的应用前景。GAN 本质上是由两个相互竞争的神经网络(生成器和判别器)组成的系统。

    机器之心
  • CVPR 2017精彩论文解读:综合使用多形态核磁共振数据的3D生物医学图像分割方法 | 分享总结

    论文的故事还在继续 相对于 CVPR 2017收录的共783篇论文,即便雷锋网(公众号:雷锋网) AI 科技评论近期挑选报道的获奖论文、业界大公司论文等等是...

    AI研习社
  • Eclipse CDT launch failed.Binary not found解决方案

    最近在研究人工智能,身份识别系统相关方面,用到c++,但是作为一个phper出身的,显然不能一帆风顺,会遇到各种坑,既然遇到了,躲不过,那就各个击破吧!

    botkenni
  • 碉堡了!一小时爬取百万知乎用户信息的Python神器曝光

    本文转载自简书,由马哥教育Python运维班3期学员推荐,原文作者为志朋,经小编编辑而成,如有漏洞,欢迎指正,并最后致谢作者的辛苦付出。 知乎是一个真实的网络...

    小小科
  • 微服务架构:自动扩展简介

    在本文中,我们将注意力集中在动态缩放,即自动扩展,以及为什么我们需要可以自动扩展的应用程序。

    sanshengshui
  • wordpressDUX主题首页缩略图无图时随机图片方法

    今天全百科网给大家分享的是wordpressDUX主题首页文章特色图片,如果在文章内容无图或未开启第一张作为特色图片时则自动调用random目录的随便一张图片作...

    于飞云计算
  • Excel常用数据导入方法

    今天给大家讲解Excel数据源的导入 ▽ excel支持的数据源类型有很多 今天只讲解常用的三种类型 Access文件、网页数据、文本数据 Access数据源导...

    数据小磨坊
  • CRM WebClient UI和Hybris里工作中心跳转的url生成逻辑

    把Work center的navigation target在client side不可见:在Chrome development tool里看不到,而是点击了...

    Jerry Wang

扫码关注云+社区

领取腾讯云代金券

玩转腾讯云 有奖征文活动