TensorFlow Plugin API reference¶
-
nvidia.dali.plugin.tf.
DALIIterator
()¶ DALI TensorFlow plugin
Creates a Dali pipeline for classification tasks from serialized DALI pipeline (given in serialized_pipeline parameter). Returns 2 TensorFlow tensors with data and label. shape must match the first DALI Pipeline output tensor shape (must be dim 4, either NCHW or NHWC). image_type must match the type of the first DALI Pipeline output tensor. label_type must match the type of the second DALI Pipeline output tensor.
Parameters: - serialized_pipeline – A string.
- shape – A tf.TensorShape or list of ints.
- num_threads – An optional int. Defaults to -1.
- device_id – An optional int. Defaults to -1.
- image_type – An optional tf.DType from: tf.float32, tf.int32, tf.int64, tf.half. Defaults to tf.float32.
- label_type – An optional tf.DType from: tf.float32, tf.int32, tf.int64. Defaults to tf.int64.
- name – A name for the operation (optional).
Returns: A tuple of Tensor objects (batch, label).
batch: A Tensor of type image_type. label: A Tensor of type label_type.