英伟达开源数据增强和数据解码库
极客时间编辑部
讲述:杜力大小:998.19K时长:02:08
在最近的 CVPR 2018 大会上,英伟达宣布,开源其数据增强和数据解码的两个新的库。其中,NVIDIA DALI 是一个 GPU 加速的数据增强和图像加载库,而 NVIDIA nvJPEG 是一个高性能的 GPU 加速的图像解码库。
据介绍,通过使用 GPU 加速数据增强,NVIDIA DALI 解决了当今计算机视觉深度学习应用程序中的性能瓶颈问题,这些应用程序包括复杂的多阶段数据增强步骤。
具体而言,DALI 的优势包括:
能够与 MXNet,TensorFlow 和 PyTorch 的框架直接集成;
具有支持多种数据格式(如 JPEG,原始格式,LMDB,RecordIO 和 TFRecord)的便携式训练工作流;
具有可配置图形和自定义操作符的自定义数据 pipeline;
使用 nvJPEG 进行高性能的单图像解码和批量 JPEG 解码。
另外,nvJPEG 是一个用于 JPEG 解码的高性能 GPU 加速库。它支持同时使用 CPU 和 GPU 对单个图像和批量图像进行解码、颜色空间转换、多相位解码,以及混合解码。与只使用 CPU 解码相比,使用 nvJPEG 进行解码的应用程序,具有更高的吞吐量和更低的延迟。
据悉,nvJPEG 可以为常用的 JPEG 格式提供低延迟解码器,如图像分类、物体检测和图像分割等应用。对于深度学习训练应用,它可以加速数据加载和预处理,如平移、缩小,放大,翻转等。对于需要低延迟深度学习推理的应用,它可以实时地进行 JPEG 解码和调整大小。
总体来看,nvJPEG 的具体优势为:
使用 CPU 和 GPU 进行混合解码;
可以进行单个图像解码和批量图像解码;
色彩空间可以转换为 RGB、BGR、RGBI 和 YUV 等;
进行单相位解码和多相位解码。
公开
同步至部落
取消
完成
0/2000
荧光笔
直线
曲线
笔记
复制
AI
- 深入了解
- 翻译
- 解释
- 总结
该免费文章来自《极客视点》,如需阅读全部文章,
请先领取课程
请先领取课程
免费领取
© 版权归极客邦科技所有,未经许可不得传播售卖。 页面已增加防盗追踪,如有侵权极客邦将依法追究其法律责任。
登录 后留言
全部留言(1)
- 最新
- 精选
- 识空屋图形解码与深度学习,mark
收起评论