NVIDIA发布最新数据增强库和图像解码库

AiTechYun

编辑:chux

今天在计算机视觉和模式识别(CVPR)会议上,NVIDIA公开了一系列新动向。

NVIDIA DALI和NVIDIA nvJPEG

首先是提供了用于数据增强和图像解码的新库。

  • NVIDIA DALI:GPU加速数据增强和图像加载库,用于优化深度学习框架的数据管道
  • NVIDIA nvJPEG:用于JPEG解码的高性能GPU加速库

由深度学习支持的计算机视觉应用包括复杂的多阶段预处理数据流水线,其中包括计算密集型步骤,如从磁盘加载和提取数据,解码,裁剪和调整大小,颜色和空间变换以及格式转换。

通过使用GPU加速数据增强,NVIDIA DALI解决了当今计算机视觉深度学习应用中的性能瓶颈问题,其中包括复杂的多阶段数据增强步骤。借助DALI,深度学习研究人员可以通过MXNet,TensorFlow和PyTorch跨越亚马逊Web服务P3 8 GPU实例或带Volta GPU的DGX-1系统,在图像分类模型(如ResNet-50)上调整训练性能。由于跨框架的一致的高性能数据加载和增强,框架用户将会减少代码重复。

DALI依靠新的NVIDIA nvJPEG库进行高性能GPU加速解码。nvJPEG支持使用CPU和GPU对单个和批量图像进行解码,色彩空间转换,多阶段解码以及混合解码。与纯CPU解码相比,依赖nvJPEG进行解码的应用,达到更高的吞吐量和更低的延迟JPEG解码。

DALI的优势包括:

  • 简单的框架与MxNet,TensorFlow和PyTorch的直接插件集成
  • 具有多种支持的数据格式(如JPEG,原始格式,LMDB,RecordIO和TFRecord)的便携式训练工作流程
  • 使用可配置图形和自定义运算符定制数据管道
  • 使用nvJPEG的高性能单一和批量JPEG解码

nvJPEG的优势包括:

  • 使用CPU和GPU进行混合解码
  • 单一图像和批量图像解码
  • 色彩空间转换为RGB,BGR,RGBI,BGRI和YUV
  • 单相和多相解码

DALI是开源的,现在可在GitHub上使用。

Apex

NVIDIA也展示了Apex的初期版本,这是一款开源的PyTorch扩展,可帮助用户最大限度地提高NVIDIA Volta GPU上的深度学习训练性能。

灵感来源于翻译网络,情感分析和图像分类方面的最新技术,NVIDIA PyTorch开发人员已经创建了将这些方法带到各级PyTorch用户的工具。Apex中的混合精密实用程序旨在提高训练速度,同时保持单精度训练的准确性和稳定性。具体而言,Apex提供FP16或FP32操作的自动执行,主参数转换的自动处理以及自动损失调整。

TensorRT 4

TensorRT 4,具有加速流行推理应用的功能,如神经机器翻译,推荐系统和语音。你还可以通过ONNX格式轻松地从流行的深度学习框架(如Caffe 2,Chainer,MxNet,Microsoft Cognitive Toolkit和PyTorch)导入模型。

原文发布于微信公众号 - ATYUN订阅号(atyun_com)

原文发表时间:2018-06-20

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

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏机器之心

业界 | Apache MXNet 发布 v0.11.0版,支持苹果Core ML和Keras v1.2

选自AWS 机器之心编译 参与:黄小天、路雪 近日,孵化于 Apache 软件基金会 Apache MXNet 发布了 v0.11.0 最终测试版本,它不仅具有...

32413
来自专栏吉浦迅科技

【讲座】在NVIDIA Jetson上从Tensorflow到TensorRT

NVIDIA在太平洋时间3月8日上午11:00-12:00(北京时间3月9日凌晨3:00-4:00)举办了主题为“AI at the Edge: TensorF...

5736
来自专栏深度学习之tensorflow实战篇

IBM SPSS Modeler Social Network Analysis 的介绍和日常应用

IBM SPSS Modeler Social Network Analysis,中文叫做社交网络分析,本文将一律简称 SNA。 引言 IBM Business...

31110
来自专栏AI科技评论

深度| OpenAI 教你如何构建深度学习研究的基础设施

编者按:OpenAI研究工程师Vicki Cheung, Jonas Schneider , Ilya Sutskever, and Greg Brockman...

3766
来自专栏AlgorithmDog的专栏

拖拽式机器学习的爱与恨

拖拽式机器学习是我想了很久的问题。 ? 1. 前世今生 拖拽式机器学习是,人们在界面上通过拖拽就是建立机器学习过程。拖拽式机器学习系统一...

3786
来自专栏ATYUN订阅号

2018年9月机器学习Github开源项目TOP 10

在此期间,我们将项目与新的或主要版本进行了比较。Mybridge AI根据各种因素对项目进行排名,以衡量专业人员的质量。

1133
来自专栏程序你好

Tensorflow实现在浏览器的深度学习

1113
来自专栏量子位

搞自动驾驶汽车很复杂?其实一个浏览器就行(讲解、代码全都有)

作者:janhuenermann 编译:量子位(QbitAI) · 问耕 自动驾驶高大上?其实在浏览器里也能玩。 上面那个视频是一个2D环境下的无人驾驶系统,...

3105
来自专栏AI科技评论

深度 | 基于移动设备的机器学习,本地与云端孰优孰劣?

AI科技评论按:如果您觉得,是时候给自己的手机应用添加一些热门的机器学习或深度学习算法.....这是个好想法!但您会怎么选择?致力于提供算法服务及小白科普的咨询...

3186
来自专栏IT大咖说

自主研发、不断总结经验,美团搜索推荐机器学习平台

内容来源:2018 年 5 月 26 日,美团点评技术专家杨一帆在“饿了么技术沙龙·第25弹【搜索推荐】”进行《Why WAI: 美团点评搜索推荐机器学习平台》...

2785

扫码关注云+社区

领取腾讯云代金券