nvidia.dali.fn.readers#

此模块中的运算符是数据生成运算符,它们从存储或不同的来源读取数据,并且数据位置在 pipeline 构建时通过参数已知。对于能够从通过常规输入动态指定的源读取数据的 data reader,请参阅 nvidia.dali.fn.io 模块。

下表列出了 nvidia.dali.fn.readers 模块中所有可用的操作

函数

设备支持

简短描述

readers.caffe

CPU

从 Caffe LMDB 读取 (图像,标签) 对。

readers.caffe2

CPU

从 Caffe2 Lightning Memory-Mapped Database (LMDB) 读取样本数据。

readers.coco

CPU

从 COCO 数据集读取数据,该数据集由包含图像和注释 JSON 文件的目录组成。

readers.file

CPU

读取文件内容并返回文件-标签对。

readers.mxnet

CPU

从 MXNet RecordIO 读取数据。

readers.nemo_asr

CPU

从 NVIDIA NeMo 兼容的 manifest 读取自动语音识别 (ASR) 数据 (音频,文本)。

readers.numpy

CPU, GPU

从目录读取 Numpy 数组。

readers.sequence

CPU

从目录读取 [帧] 序列,该目录表示流的集合。

readers.tfrecord

CPU

从 TensorFlow TFRecord 文件读取样本。

readers.video

GPU

使用 FFmpeg 和 NVDECODE 加载和解码视频文件,NVDECODE 是 NVIDIA(R) GPU 中的硬件加速视频解码功能。

readers.video_resize

GPU

使用 FFmpeg 和 NVDECODE 加载、解码和调整视频文件大小,NVDECODE 是 NVIDIA GPU 的硬件加速视频解码。

readers.webdataset

CPU

用于 webdataset 格式的 reader。